coordinate(alongAxisTensor:withShapeTensor:name:)
Creates a get-coordindate operation and returns the result tensor.
Declaration
func coordinate(alongAxisTensor axisTensor: MPSGraphTensor, withShapeTensor shapeTensor: MPSGraphTensor, name: String?) -> MPSGraphTensorParameters
- axisTensor:
A Scalar tensor of type
MPSDataTypeInt32, that specifies the coordinate axis an element’s value is set to. Negative values wrap around. - shapeTensor:
A rank-1 tensor of type
MPSDataTypeInt32orMPSDataTypeInt64that defines the shape of the result tensor. - name:
The name for the operation.
Return Value
A valid MPSGraphTensor object.