torch.fmax

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

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

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

Эта функция является обёрткой над std::fmax из C++ и похожа на функцию fmax из NumPy.

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

Параметры:

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

Именованные аргументы:

out (Tensor, optional) – выходной тензор.

Пример:

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