torch.clamp()简要介绍

定义

torch.clamp(input,min,max)

用法

将输入input每个元素的值限制在min和max之间，小于min或者大于max的值分别替换为min和max

示例

import torch

a = torch.tensor([[2,3,1],[0,5,6]])
b = torch.clamp(a,2,5)
print(a)
print(b)

>>>tensor([[2, 3, 1],
           [0, 5, 6]])

tensor([[2, 3, 2],
        [2, 5, 5]])