假設 1個影像樣本由神經網路處理后的輸出是 size 10×4 的tensor，隨機生成一個tensor，
使用Sigmoid對該tensor進行概率變換，tensor的每個數值的變換都是相互獨立的，下面得出預測的概率分布，
我們在這里隨機生成一個真值的概率分布y，
自己定義一下BCELoss，需要輸入預測的概率分布和真值的概率分布，
手動計算一下，結果一致，
使用F.binary_cross_entropy 驗證一下，同樣需要輸入預測的概率分布和真值的概率分布，
再使用F.binary_cross_entropy_with_logits驗證一下，直接輸入神經網路的輸出和真值的概率分布，
BCELoss與BCEWithLogitsLoss的關聯：BCEWithLogitsLoss = Sigmoid + BCELoss，
nn介面 Function介面
nn.BCELoss( ) F.binary_cross_entropy( )
nn.BCEWithLogitsLoss( ) F.binary_cross_entropy_with_logits( )
Enjoy it!
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/551211.html
標籤：其他
上一篇：牛客小白月賽71
下一篇：返回列表