torch.minimum

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

Вычисляет поэлементное минимальное значение input и other.

Примечание

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

Параметры

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

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

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

Пример:

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