Pytorch.clamp：将小于0的元素修改为0，截断元素的取值空间

Pytorch.clamp：将小于0的元素修改为0，截断元素的取值空间

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

将输入中每个元素夹紧到[min,max]区间中，算法公式如下：
$$y_i=\{\begin{array}{ll}\min & \text{ if }{x}_i<\min \\ x_i& \text{ if }\min \le x_i\le \max \\ \max & \text{ if }{x}_i>\max \end{array}$$
例：将小于0的元素修改为0

a = torch.randn(5)
# tensor([-0.8286,  0.4988, -0.5804,  1.0583,  1.6374])
torch.clamp(a,min=0.0)
# tensor([0.0000, 0.4988, 0.0000, 1.0583, 1.6374])

非常好用~
参考：

1. pytorch:torch.clamp()