megengine.functional.argmin

argmin(inp, axis=None, keepdims=False)[源代码]

返回在给定轴 axisinp 张量每一行的最小值的下标。如果给定的轴是一个维度列表,则规约所有维度。

参数
  • inp – 输入张量。

  • axis – 要进行规约的维度。如果设置为 None,则所有的维度都将规约。默认:None

  • keepdims – 输出张量是否保留了轴 axis 。默认:False

返回

输出张量。

例如:

import numpy as np
from megengine import tensor
import megengine.functional as F

x = tensor(np.arange(1, 7, dtype=np.int32).reshape(2,3))
out = F.argmin(x)
print(out.numpy())

输出:

0