torch.clamp()

2022-09-02 22:26:02 浏览数 (2)

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

将输入input张量每个元素的范围限制到区间 [min,max],返回结果到一个新张量。

  • input (Tensor) – 输入张量
  • min (Number) – 限制范围下限
  • max (Number) – 限制范围上限
  • out (Tensor, optional) – 输出张量

例如:

代码语言:javascript复制
a=torch.randint(low=0,high=10,size=(10,1))
print(a)
a=torch.clamp(a,3,9)
print(a)

输出为:

0 人点赞