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)