torch.clamp()

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])