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