torch.fmin

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

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

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

Эта функция является оболочкой вокруг C++ функции std::fmin и аналогична функции NumPy 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])