torch.fmax

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

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

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

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

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

Параметры

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

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

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

Пример:

>>> a = torch.tensor([9.7, float('nan'), 3.1, float('nan')])
>>> b = torch.tensor([-2.2, 0.5, float('nan'), float('nan')])
>>> torch.fmax(a, b)
tensor([9.7000, 0.5000, 3.1000,    nan])