torch.fmin¶
- torch.fmin(input, other, *, out=None) Tensor ¶
计算
input
和other
的元素级最小值。这类似于
torch.minimum()
,但对 NaN 的处理方式不同:如果比较的两个元素中只有一个是 NaN,则取非 NaN 的元素作为最小值。只有当两个元素都是 NaN 时,才会传播 NaN。此函数是 C++
std::fmin
的包装器,与 NumPy 的fmin
函数类似。- 参数
- 关键字参数
out (Tensor, optional) – 输出 Tensor。
示例
>>> 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])