torch.clamp()

1.用法示例

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

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

当然，需要注意的在于，input张量的dtype和[min,max]的数据类型
保持一致？？

数学表示如下：

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

Example:

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

结果：

torch.clamp()和np.clip()应该是差不多的

2.clamp_下划线

有下划线的表示修改并付给自身，无下划线的表示需要返回处理后的值

h = k.clamp(min=0)　＃将结果存入ｈ，ｋ保留原值

k.clamp_(min=0)   # 将结果存入ｋ

3.参考

1.pytorch:torch.clamp()
2.pytorch clamp与clamp_区别