mindspore.ops.fmax
- mindspore.ops.fmax(input, other)[源代码]
逐元素计算输入tensor的最大值。
说明
支持隐式类型转换、类型提升。
输入 input 和 other 的shape必须能相互广播。
如果其中一个比较值是NaN,则返回另一个比较值。
- 参数:
input (Tensor) - 第一个输入。
other (Tensor) - 第二个输入。
- 返回:
Tensor
- 支持平台:
CPU
样例:
>>> import mindspore >>> x1 = mindspore.tensor([1.0, 5.0, 3.0], mindspore.float32) >>> x2 = mindspore.tensor([4.0, 2.0, 6.0], mindspore.float32) >>> output = mindspore.ops.fmax(x1, x2) >>> print(output) [4. 5. 6.]