Pytorch詳解NLLLoss和CrossEntropyLoss

pytorch的官方文檔寫的也太簡陋了吧…害我看了這麼久… NLLLoss 在圖片單標籤分類時，輸入m張圖片，輸出一個m*N的Tensor，其中N是分類個數。比如輸入3張圖片，分三類，最後的輸出是一個3*3的Tensor，舉個例子： 第123行分別是第123張圖片的結果，假設第123列分別是貓、狗和豬的分類得分。 可以看出模型認爲第123張都更可能是貓。 然後對每一行使用Softmax，這樣可以得

>>阅读原文<<