将输入的张量中的所有值限定在某个范围中
Parameters:
- input: 输入的张量
- min: 范围下界
- max: 范围上界
如果输入的张量中有值小于下界,则将其替换为下界值,上界同理
a = torch.randn(4)
a
tensor([-1.7120, 0.1734, -0.0478, -0.0922])
torch.clamp(a, min=-0.5, max=0.5)
min = torch.linspace(-1, 1, steps=4)
torch.clamp(a, min=min)
tensor([-1.0000, 0.1734, 0.3333, 1.0000])