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
'''