megengine.functional.min¶

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

计算tensor在给定轴（或多个轴）上的最小值。

参数

inp (Tensor) – 输入张量。应该有一个数字数据类型。
axis (Union[int, Sequence[int], None]) – 指定计算最小值的一个或多个轴。默认情况下，是整个tensor上计算最小值。如果是整数序列，会在多个轴上计算最小值。
keepdims (bool) – 若为 “True”，则指定用来reduce的轴(维度)必须包含在结果中，因此，结果必须与输入tensor兼容(参见:ref:`broadcasting-rule `)。否则，若为 “False”，结果中不得包含用于reduce的轴(维度)。

返回类型

Tensor

返回

如果是计算整个 tensor 的最小值，则是包含最小值的零维tensor；否则，包含最小值的非零维tensor。

特殊情况

如果 \(x_i\) 是 NaN，最小值是 NaN``(即传播 ``NaN)。

实际案例

>>> x = Tensor([[1, 2], [3, 4]])
>>> F.min(x)
Tensor(1, dtype=int32, device=xpux:0)

沿着某一轴：

>>> F.min(x, axis=0)
Tensor([1 2], dtype=int32, device=xpux:0)
>>> F.min(x, axis=1)
Tensor([1 3], dtype=int32, device=xpux:0)

megengine.functional.mean

megengine.functional.max