Contents

gatherAlongAxis(_:updates:indices:name:)

Creates a GatherAlongAxis operation and returns the result tensor.

Declaration

func gatherAlongAxis(_ axis: Int, updates updatesTensor: MPSGraphTensor, indices indicesTensor: MPSGraphTensor, name: String?) -> MPSGraphTensor

Parameters

  • axis:

    The axis to gather from. Negative values wrap around

  • updatesTensor:

    The input tensor to gather values from

  • indicesTensor:

    Int32 or Int64 tensor used to index updatesTensor

  • name:

    The name for the operation.

Return Value

A valid MPSGraphTensor object

Discussion

Gather values from updatesTensor along the specified axis at indices in indicesTensor. The shape of updatesTensor and indicesTensor must match except at axis. The shape of the result tensor is equal to the shape of indicesTensor. If an index is out of bounds of the updatesTensor along axis a 0 is inserted.