coordinate(alongAxisTensor:withShape:name:)
Creates a get-coordindate operation and returns the result tensor.
Declaration
func coordinate(alongAxisTensor axisTensor: MPSGraphTensor, withShape shape: [NSNumber], 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. - shape:
The shape of the result tensor.
- name:
The name for the operation.
Return Value
A valid MPSGraphTensor object.