torch.fmin

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

Вычисляет поэлементный минимум input и other.

Работает подобно torch.minimum(), но иначе обрабатывает NaN: если только один из двух сравниваемых элементов является NaN, минимумом считается элемент, не являющийся NaN. NaN распространяется только в том случае, если оба элемента являются NaN.

Эта функция является обёрткой для std::fmin в C++ и аналогична функции fmin в NumPy.

Поддерживает широковещательное преобразование к общей форме, повышение типа, а также целочисленные данные и числа с плавающей точкой.

Параметры:

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