torch.clamp

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

将 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 的值。

Parameters

• 输入 (张量) – 输入张量。

• 最小值 (数值 或 张量, 可选) – 要钳制到的范围的下界

• 最大值 (数字 或 张量，可选) – 要钳制的范围的上限

Keyword Arguments

输出 (张量, 可选) – 输出张量。

示例：

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