pytorch笔记 torch.clamp（截取上下限）

1 基本用法

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

使得tensor中比min小的变成min，比max大的变成max

2 使用举例

import torch
a = torch.randn(4)
print(a)
#tensor([-1.7120,  0.1734, -0.0478, -0.0922])

torch.clamp(a, min=-0.5, max=0.5)
#tensor([-0.5000,  0.1734, -0.0478, -0.0922])