mindspore.Tensor.argmax
- mindspore.Tensor.argmax(axis=None)[源代码]
返回指定轴上最大值的索引。
- 参数:
axis (int, 可选) - 默认情况下,返回扁平化Tensor的最大值序号,否则返回指定轴方向上。
- 返回:
Tensor,最大值的索引。它与原始Tensor具有相同的shape,但移除了轴方向上的维度。
- 异常:
ValueError - 入参axis的设定值超出了范围。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import numpy as np >>> from mindspore import Tensor >>> a = Tensor(np.arange(10, 16).reshape(2, 3).astype("float32")) >>> print(a.argmax()) 5