torch.maximum

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

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

Примечание

Если один из сравниваемых элементов является NaN, возвращается этот элемент. maximum() не поддерживается для тензоров с комплексными типами данных.

Параметры:

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

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

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

Пример:

>>> a = torch.tensor((1, 2, -1))
>>> b = torch.tensor((3, 0, 4))
>>> torch.maximum(a, b)
tensor([3, 2, 4])