以前我瀏覽博客的時(shí)候記得別人說過，BCELoss與CrossEntropyLoss都是用于分類問題。可以知道，BCELoss是Binary CrossEntropyLoss的縮寫，BCELoss　CrossEntropyLoss的一個(gè)特例，只用于二分類問題，而CrossEntropyLoss可以用于二分類，也可以用于多分類。

不過我重新查閱了一下資料，發(fā)現(xiàn)同樣是處理二分類問題，BCELoss與CrossEntropyLoss是不同的。下面我詳細(xì)講一下哪里不同。

１．　使用nn.BCELoss需要在該層前面加上Sigmoid函數(shù)。
公式如下：
$$loss(X_i,y_i)=?w_i[y_{i}log{x}_i+(1?y_i)log(1?x_i)]$$
２．　使用nn.CrossEntropyLoss會(huì)自動(dòng)加上Sofrmax層。
公式如下：
$$loss(X_i,y_i)=?w_{label}log\frac{e^{x_{label}}}{{\sum }_{j=1}^N{e}^{x_j}}$$

可以看出，這兩個(gè)計(jì)算損失的函數(shù)使用的激活函數(shù)不同，故而最后的計(jì)算公式不同。

總結(jié)

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

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