reductionArgMaximum(with:axis:name:)
Creates a reduction argMax operation and returns the result tensor.
Declaration
func reductionArgMaximum(with tensor: MPSGraphTensor, axis: Int, name: String?) -> MPSGraphTensorParameters
- tensor:
Input tensor
- axis:
Axis of reduction
- name:
Name for the operation
Return Value
A valid MPSGraphTensor object.