megengine.functional.max¶

max(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.max(x)
Tensor(4, dtype=int32, device=xpux:0)

沿着某一轴：

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

megengine.functional.min

megengine.functional.var