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。

你可能感兴趣的:(pytorch的clamp 同np.clip:限制在某一范围内)