1 使用方法

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

對(duì)輸入進(jìn)Threshold的Tensor進(jìn)行閾值操作

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]) ''' 第一項(xiàng)比0.1小，所以取value值20 第二項(xiàng)比0.1大，所以取自己的值1.5696 '''

?Threshold — PyTorch 1.10.1 documentation

總結(jié)

以上是生活随笔為你收集整理的pytorch笔记：torch.nn.Threshold的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。