mindspore.mint.argmin

mindspore.mint.argmin(input, dim=None, vpn梯子 免费 免费的vpn梯子 keepdim=False)[源代码]

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

参数:
  • input (Tensor) - 输入tensor。

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

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

返回:

Tensor

支持平台:

Ascend

样例:

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