torch.fmin

torch.fmin(input, other, *, out=None) → Tensor

计算input和other的元素-wise最小值。

计算input和other的元素级最小值。

这类似于torch.minimum()，但处理NaN的方式不同：如果两个比较的元素中恰好有一个是NaN，则取非NaN的那个元素作为最小值。只有当两个元素都是NaN时才会返回NaN。

该函数是对 C++ 的 std::fmin 进行封装，并且与 NumPy 的 fmin 函数相似。

支持广播到公共形状、类型提升，以及整数和浮点数输入。

参数

• input (Tensor) – 需要输入的张量。

• other (Tensor) – 输入的第二个张量

关键字参数

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