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 输出张量
例子:
在这里插入图片描述

你可能感兴趣的:(深度学习,pytorch,深度学习,人工智能)