1 使用方法

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

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

2 使用举例 

import torcha=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

总结

以上是生活随笔为你收集整理的pytorch笔记：torch.nn.Threshold的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。