megengine.functional.maximum

maximum(x, y)[源代码]

Tensor中的逐元素最大值。

逐元素比较两个 tensor 的最大值,返回新 tensor。如果与”NaN”进行比较,则返回该元素。如果用于比较的两个元素都是”NaNs”,则返回第一个元素。

参数
  • x – 输入tensor。 类型为数值数据类型。

  • y – 输入tensor。 类型与:math:`x`的数据类型一致。

返回

包含最大元素的tensor。返回tensor的数据类型必须与:math:`x`的数据类型一致。

实际案例

>>> F.maximum(1, 2)
Tensor(2, dtype=int32, device=xpux:0)

逐元素求最大值:

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

广播:

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