coordinate(alongAxis:withShapeTensor:name:)
Creates a get-coordindate operation and returns the result tensor.
Declaration
func coordinate(alongAxis axis: Int, withShapeTensor shapeTensor: MPSGraphTensor, name: String?) -> MPSGraphTensorParameters
- axis:
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.