Python
torch.clamp()
函数原型:
torch.clamp(input, min=None, max=None, *, out=None)
clamp的是夹子,夹紧
所以这个函数的作用是将输入的input夹紧在min和max之间
当input在min和max之间的时候输出本身,小于min的时候输出min 大于max时,输出max
import torch
a = torch.randn(4)
b = torch.clamp(a, min=-0.5, max=0.5)
print(a, '\n', b)
输出为:
tensor([-1.7120, 0.1734, -0.0478, -0.0922])
tensor([-0.5000, 0.1734, -0.0478, -0.0922])