每天学习一个Torch函数之torch.clamp

函数原型

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

理解

clamp的英文意思是”夹子、夹紧“，英英解释：”A clamp is a device that holds two things firmly together.“
所以这个函数的作用是将输入的Tensor“夹”在最小值和最大值之间。
数学式为：
$$y_i=min(max(x_i,min\_valu{e}_i),max\_valu{e}_i)$$

参数

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