mindspore.mint.argmin

mindspore.mint.argmin(input, dim=None, keepdim=False)

返回tensor在指定维度上的最小值索引。

参数：

• input (Tensor) - 输入tensor。

• dim (Union[int, None]，可选) - 指定维度。如果为 None ，计算 input 中的所有元素。默认 None 。

• keepdim (bool，可选) - 输出tensor是否保留维度。默认 False 。

返回：

Tensor

支持平台：

Ascend

样例：

>>> import mindspore
>>> import numpy as np
>>> x = mindspore.tensor(np.array([[1, 20, 5], [67, 8, 9], [130, 24, 15]]), mindspore.float32)
>>> output = mindspore.mint.argmin(x, dim=-1, keepdim=False)
>>> print(output)
[0 1 2]