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) #将结果存入h,k保留原值
k.clamp_(min=0) # 将结果存入k