当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch clamp用法及代码示例


本文简要介绍python语言中 torch.clamp 的用法。

用法:

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

参数

  • input(Tensor) -输入张量。

  • min(数字或者Tensor,可选的) -lower-bound 要钳制到的范围

  • max(数字或者Tensor,可选的) -upper-bound 要钳制到的范围

关键字参数

out(Tensor,可选的) -输出张量。

input 中的所有元素夹在 [ min max ] 范围内。让 min_value 和 max_value 分别为 min max ,这将返回:

如果 min None ,则没有下限。或者,如果 max None,则没有上限。

注意

如果 min 大于 max torch.clamp(..., min, max)input 中的所有元素设置为 max 的值。

例子:

>>> a = torch.randn(4)
>>> a
tensor([-1.7120,  0.1734, -0.0478, -0.0922])
>>> torch.clamp(a, min=-0.5, max=0.5)
tensor([-0.5000,  0.1734, -0.0478, -0.0922])

>>> min = torch.linspace(-1, 1, steps=4)
>>> torch.clamp(a, min=min)
tensor([-1.0000,  0.1734,  0.3333,  1.0000])

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.clamp。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。