PyTorch torch.fmax 函数
Pytorch torch 参考手册torch.fmax 是 PyTorch 中用于逐元素取最大值的函数（忽略NaN）。当输入中包含NaN时，torch.maximum会返回NaN，而torch.fmax会忽略NaN并返回非NaN值。
函数定义
torch.fmax(input, other)
使用示例
实例
import torch
# 创建包含NaN的张量
x = torch.tensor([3.0, float('nan'), 4.0])
y = torch.tensor([1.0, 5.0, 2.0])
# 逐元素取最大值（忽略NaN）
result = torch.fmax(x, y)
print(result)
# 创建包含NaN的张量
x = torch.tensor([3.0, float('nan'), 4.0])
y = torch.tensor([1.0, 5.0, 2.0])
# 逐元素取最大值（忽略NaN）
result = torch.fmax(x, y)
print(result)
输出结果为：
tensor([3., 5., 4.])