pytorch的clamp 同np.clip:限制在某一范围内

1.numpyclip

numpy.clip(a, a_min, a_max, out=None)

a是一个数组,后面两个参数分别表示最小和最大值。
也就是说clip这个函数将将数组中的元素限制在a_min, a_max之间,大于a_max的就使得它等于 a_max,小于a_min,的就使得它等于a_min。

import numpy as np
x=np.array([1,2,3,5,6,7,8,9])
np.clip(x,3,8)
Out[1]:
array([3, 3, 3, 5, 6, 7, 8, 8])

2.pytorch的clamp

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

clamp函数clamp表示夹紧,夹住。将input中的元素限制在[min,max]范围内并返回一个Tensor。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值