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的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。