torch.clamp_()函数

1.clamp()函数:

    参考资料:https://blog.csdn.net/jacke121/article/details/85270621

    1)函数功能:

          clamp()函数的功能将输入input张量每个元素的值压缩到区间 [min,max],并返回结果到一个新张量。

    2)参数列表:

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

          input:输入张量;

          min:限制范围下限;

          max:限制范围上限;

          out:输出张量。 

    3)应用举例:

a=torch.randint(low=0,high=10,size=(10,1))
print(a)
b=torch.clamp(a,3,9)
print(b)

 

2.clamp_()与clamp()的区别:

    参考资料:https://blog.csdn.net/qq_27261889/article/details/86777230

     pytorch中,一般来说如果对tensor的一个函数后加上了下划线,则表明这是一个in-place类型。in-place类型是指,当在一个tensor上操作了之后,是直接修改了这个tensor,而不是返回一个新的tensor并不修改旧的tensor。

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐