torch.clamp()函数 用于标签平滑

用于标签平滑

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

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

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

例

>>a = torch.Tensor([[1, 0, 0], 
                    [0, 1, 0], 
                    [0, 0, 1]])
>>b = torch.clamp(a, 0.1, 0.9)
>>print(b)
>>tensor([[0.9, 0.1, 0.1], 
          [0.1, 0.9, 0.1], 
          [0.1, 0.1, 0.9]])