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)