mindspore.numpy.argmax
- mindspore.numpy.argmax(a, axis=None)[source]
Returns the indices of the maximum values along an axis.
Note
Numpy argument out is not supported. On Ascend, in case of multiple occurrences of the maximum values, the return indices may not necessarily correspond to the first occurrence.
- Parameters
- Returns
Tensor, array of indices into the array. It has the same shape as a.shape with the dimension along axis removed.
- Raises
ValueError – if axis is out of range.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> 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]