首页 > 其他 > 详细

torch.clamp() 函数

时间:2020-09-05 13:47:47      阅读:578      评论:0      收藏:0      [点我收藏+]

 

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

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

操作定义如下:

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

 

a=torch.randint(low=0,high=10,size=(10,1))
print(a)
a=torch.clamp(a,3,9)
print(a)
tensor([[2],
        [1],
        [3],
        [0],
        [0],
        [4],
        [3],
        [6],
        [0],
        [5]])
tensor([[3],
        [3],
        [3],
        [3],
        [3],
        [4],
        [3],
        [6],
        [3],
        [5]])

torch.clamp() 函数

原文:https://www.cnblogs.com/tingtin/p/13617764.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!