mindspore.ops.Argmax
- class mindspore.ops.Argmax(axis=- 1, output_type=mstype.int32)[source]
Returns the indices of the maximum value along a specified axis of a Tensor.
Refer to
mindspore.ops.argmax()
for more details.- Parameters
axis (int) – Axis where the Argmax operation applies to. Default:
-1
.output_type (
mindspore.dtype
) – Output data type. Supported types:mstype.int32
,mstype.int64
. Default:mstype.int32
.
- Inputs:
input_x (Tensor) - The input tensor. \((N, *)\) where \(*\) means, any number of additional dimensions.
- Outputs:
Tensor, indices of the max value of input tensor across the axis.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> input_x = Tensor(np.array([[1, 20, 5], [67, 8, 9], [130, 24, 15]]).astype(np.float32)) >>> output = ops.Argmax(output_type=mindspore.int32)(input_x) >>> print(output) [1 0 0]