torch.fmin

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

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

Это похоже на torch.minimum(), за исключением того, как обрабатываются NaN: если ровно один из двух сравниваемых элементов — NaN, то не-NaN элемент принимается как минимум. Только если оба элемента являются NaN, NaN передаётся.

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

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

Параметры

• input (Тензор) – входной тензор.
• other (Тензор) – второй входной тензор

Ключевые аргументы

out (Тензор, необязательно) – выходной тензор.

Пример:

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