Pytorch詳解NLLLoss和CrossEntropyLoss

NLLLoss 在圖片單標籤分類時，輸入m張圖片，輸出一個m*N的Tensor，其中N是分類個數。比如輸入3張圖片，分三類，最後的輸出是一個3*3的Tensor，舉個例子： 第123行分別是第123張圖片的結果，假設第123列分別是貓、狗和豬的分類得分。 可以看出模型認爲第123張都更可能是貓。 然後對每一行使用Softmax，這樣可以得到每張圖片的概率分佈。 這裏dim的意思是計算Softmax

>>阅读原文<<