pytorch中torch.clamp()函数

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

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

参数解释:

input(Tensor):输入张量
min:限制范围下限
max:限制范围上限
out(Tensor):输出张量

数学原理:

         | min, if input < min
output = | input, if min <= input <= max
         | max, if input > max
发布了26 篇原创文章 · 获赞 65 · 访问量 4321

猜你喜欢

转载自blog.csdn.net/qq_40263477/article/details/103881062