torch.clamp() 函数

 

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

将输入input张量每个元素的夹紧到区间 [min,max][min,max]，并返回结果到一个新张量。

操作定义如下：

      | min, if x_i < min
y_i = | x_i, if min <= x_i <= max
      | max, if x_i > max

 

a=torch.randint(low=0,high=10,size=(10,1))
print(a)
a=torch.clamp(a,3,9)
print(a)

tensor([[2],
        [1],
        [3],
        [0],
        [0],
        [4],
        [3],
        [6],
        [0],
        [5]])
tensor([[3],
        [3],
        [3],
        [3],
        [3],
        [4],
        [3],
        [6],
        [3],
        [5]])