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])