torch.clamp

torch.clamp(input, min=None, max=None, *, out=None) → Tensor

Ограничивает все элементы в input в диапазоне [ min, max ]. Пусть min_value и max_value будут min и max соответственно, это возвращает:

$$y_i = \min(\max(x_i, \text{min\_value}_i), \text{max\_value}_i)$$

Если min равно None, нижней границы нет. Или, если max равно None, верхней границы нет.

Примечание

Если min больше, чем max torch.clamp(..., min, max) устанавливает все элементы в input в значение max.

Параметры:

• input (Tensor) – входной тензор.
• min (Число или Tensor, необязательно) – нижняя граница диапазона, к которому необходимо применить ограничение
• max (Число или Tensor, необязательно) – верхняя граница диапазона, к которому необходимо применить ограничение

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

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

Пример:

>>> a = torch.randn(4)
>>> a
tensor([-1.7120,  0.1734, -0.0478, -0.0922])
>>> torch.clamp(a, min=-0.5, max=0.5)
tensor([-0.5000,  0.1734, -0.0478, -0.0922])

>>> min = torch.linspace(-1, 1, steps=4)
>>> torch.clamp(a, min=min)
tensor([-1.0000,  0.1734,  0.3333,  1.0000])