torch.fmax¶
- torch.fmax(input, other, *, out=None) Tensor ¶
计算
input
和other
的逐元素最大值。这类似于
torch.maximum()
,但处理 NaN 的方式不同:如果两个要比较的元素中恰好有一个是 NaN,则将非 NaN 元素作为最大值。只有当两个元素都是 NaN 时,才会传播 NaN。此函数是围绕 C++ 的
std::fmax
的包装器,类似于 NumPy 的fmax
函数。示例
>>> 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])