torch.fmin¶
- torch.fmin(input, other, *, out=None) Tensor ¶
计算
input
和other
的元素级最小值。这与
torch.minimum()
类似,但它对 NaN 的处理方式不同:如果两个要比较的元素中恰好有一个是 NaN,则将非 NaN 元素视为最小值。只有当两个元素都是 NaN 时,才会传播 NaN。此函数是 C++ 的
std::fmin
的包装器,类似于 NumPy 的fmin
函数。示例
>>> a = torch.tensor([2.2, float('nan'), 2.1, float('nan')]) >>> b = torch.tensor([-9.3, 0.1, float('nan'), float('nan')]) >>> torch.fmin(a, b) tensor([-9.3000, 0.1000, 2.1000, nan])