torch.clamp

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]])
"""