Pytorch:torch.clamp()函数

1 torch.clamp()

将输入input张量每个元素的夹紧到区间 [min,max],并返回结果到一个新张量。

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

操作定义如下:

      | min, if x_i < min
y_i = | x_i, if min <= x_i <= max
      | max, if x_i > max

如果输入是FloatTensor or DoubleTensor类型,则参数min max 必须为实数,否则须为整数。【译注:似乎并非如此,无关输入类型,min, max取整数、实数皆可。】

参数 含义
input 输入张量
min 限制范围下限
max 限制范围上限
out 输出张量

栗子:

参考:

  • https://blog.csdn.net/weixin_36670529/article/details/105964721
  • https://blog.csdn.net/jacke121/article/details/85270621

 

 

 

 

 

你可能感兴趣的:(修仙之路:pytorch篇)