megengine.functional.argmin# argmin(inp, axis=None, keepdims=False)[源代码]# 返回在给定轴 axis 上 inp 张量每一行的最小值的下标。如果给定的轴是一个维度列表,则规约所有维度。 参数: inp (Tensor) – 输入张量。 axis (Union[int, Sequence[int], None]) – 要进行规约的维度。如果设置为 None,则所有的维度都将规约。默认:None keepdims (bool) – 输出张量是否保留了轴 axis 。默认:False 返回类型: Tensor 返回: 输出张量。 实际案例 >>> import numpy as np >>> x = Tensor(np.arange(1, 7, dtype=np.int32).reshape(2,3)) >>> F.argmin(x) Tensor(0, dtype=int32, device=xpux:0)