函数原型
torch.clamp(input, min=None, max=None, *, out=None)
理解
clamp的英文意思是”夹子、夹紧“,英英解释:”A clamp is a device that holds two things firmly together.“
所以这个函数的作用是将输入的Tensor“夹”在最小值和最大值之间。
数学式为:
y
i
=
m
i
n
(
m
a
x
(
x
i
,
m
i
n
_
v
a
l
u
e
i
)
,
m
a
x
_
v
a
l
u
e
i
)
y_i=min(max(x_i, min\_value_i),max\_value_i)
yi=min(max(xi,min_valuei),max_valuei)
参数
- input: Tensor
- min: Number or Tensor
- max: Number or Tensor
返回值:out: Tensor
Example
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])