pytorch笔记：torch.nn.Threshold

1 使用方法

torch.nn.Threshold(
    threshold, 
    value, 
    inplace=False)

对输入进Threshold的Tensor进行阈值操作

2 使用举例 

import torch

a=torch.randn(2)
print((a)
#tensor([-0.6056,  1.5696])

m = torch.nn.Threshold(0.1, 20)
m(a)
#tensor([20.0000,  1.5696])
'''
第一项比0.1小，所以取value值20
第二项比0.1大，所以取自己的值1.5696
'''

 Threshold — PyTorch 1.10.1 documentation