mindspore.numpy.argmax

mindspore.numpy.argmax(a, axis=None)[源代码]

返回沿指定轴最大值的索引。

说明

不支持NumPy参数 out 。 在Ascend上，如果存在多个最大值的情况，返回的索引可能不一定对应于第一次出现的值。

参数：

  • a (Union[int, float, bool, list, tuple, Tensor]) - 输入数组。

  • axis (int, 可选) - 默认情况下，索引进入展平的数组，否则沿指定 axis 。 默认值： None 。

返回：

Tensor，原数组中元素的索引的数组。 与移除指定 axis 后的入参 a 的shape相同。

异常：

  • ValueError - 如果 axis 超出范围。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> a = np.arange(10, 16).reshape(2, 3)
>>> print(np.argmax(a))
5
>>> print(np.argmax(a, axis=0))
[1 1 1]
>>> print(np.argmax(a, axis=1))
[2 2]