clamp()函数的功能将输入input张量每个元素的值压缩到区间 [min,max],并返回结果到一个新张量。大小不超过某个界限
torch.clamp(input, min, max, out=None) → Tensor
-
input:输入张量;
-
min:限制范围下限;
-
max:限制范围上限;
-
out:输出张量。
import torch
a = torch.rand(2, 2).mul_(10)
print(a)
print(a.clamp(2, 5))
"""
out:
tensor([[4.4173, 6.0047],
[3.4808, 4.6272]])
tensor([[4.4173, 5.0000],
[3.4808, 4.6272]])
"""