1 基本用法
torch.clamp(
input,
min=None,
max=None,
*,
out=None)
使得tensor中比min小的变成min,比max大的变成max
2 使用举例
import torch
a = torch.randn(4)
print(a)
#tensor([-1.7120, 0.1734, -0.0478, -0.0922])
torch.clamp(a, min=-0.5, max=0.5)
#tensor([-0.5000, 0.1734, -0.0478, -0.0922])