megengine.Parameter.min

Parameter.min(axis=None, keepdims=False)

返回给定维度 axis 中输入张量的每一行的最小值。

如果 axis 是一个包含多个轴的列表,则会在它们上都做规约。如果 keepdimsTrue, 那么输出张量的形状保持和输入张量一致,除了 axis 指定的维度值将变为 1. 否则 axis 轴会被压缩(参考 squeeze ).

参数
  • axis – 做规约的维度或者多个维度

  • keepdims (bool) – 输出张量是否保留了轴 ndim 。

返回

输出张量。

实际案例

from megengine import tensor
a = tensor([False, True, True, False])
b = tensor([1.0, 2.0, 3.0, 4.0])
print(a.min().numpy())
print(b.min().numpy())

输出:

False
1.0