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