tensor. clamp的裁剪功能

import torch

a = torch.rand(2, 3)*10
print('a\n', a)
# 从输出结果可以看出对于a中的小于2的会将值取为2，对于大于5的也会将值改为5
# 在2和5之间的则不变，也就是说将数据处理在2-5之间
a = a.clamp(2, 5)
print('a\n', a)

输出结果

a
 tensor([[0.4248, 1.7215, 1.0914],
        [6.6450, 5.7751, 5.2326]])
a
 tensor([[2., 2., 2.],
        [5., 5., 5.]])