gather(input:indices:output:axis:filterParameters:)
Gathers the elements of a tensor along a single axis.
Declaration
static func gather(input: BNNSNDArrayDescriptor, indices: BNNSNDArrayDescriptor, output: BNNSNDArrayDescriptor, axis: Int, filterParameters: BNNSFilterParameters? = nil) throwsParameters
- input:
The input descriptor.
- indices:
The indices descriptor.
- output:
The output descriptor.
- axis:
The axis along which the operation gathers the indices.
- filterParameters:
The filter runtime parameters.
See Also
Related Documentation
Gather and scatter operations
Calculating the dominant colors in an imagegatherND(input:indices:output:filterParameters:)scatter(input:indices:output:axis:reductionFunction:filterParameters:)scatterND(input:indices:output:reductionFunction:filterParameters:)BNNSGather(_:_:_:_:_:)BNNSGatherND(_:_:_:_:)BNNSScatter(_:_:_:_:_:_:)BNNSScatterND(_:_:_:_:_:)