Contents

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) throws

Parameters

  • 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