torch.fmax

torch.fmax(input, other, *, out=None) → Tensor

计算input和other的元素-wise最大值。

计算input和other的元素级最大值。

这类似于torch.maximum()，但处理NaN的方式不同：如果两个比较的元素中只有一个为NaN，则取非NaN的那个元素作为最大值。只有当两个元素都为NaN时才会返回NaN。

此函数是对 C++ 的 std::fmax 进行封装，并且与 NumPy 的 fmax 函数相似。

支持广播到公共形状、类型提升，以及整数和浮点数输入。

参数

• input (Tensor) – 需要输入的张量。

• other (Tensor) – 输入的第二个张量

关键字参数

out (Tensor, 可选) – 指定输出张量。

示例：

>>> a = torch.tensor([9.7, float('nan'), 3.1, float('nan')])
>>> b = torch.tensor([-2.2, 0.5, float('nan'), float('nan')])
>>> torch.fmax(a, b)
tensor([9.7000, 0.5000, 3.1000,    nan])