Pytorch详解NLLLoss和CrossEntropyLoss

pytorch的官方文档写的也太简陋了吧…害我看了这么久… NLLLoss 在图片单标签分类时，输入m张图片，输出一个m*N的Tensor，其中N是分类个数。比如输入3张图片，分三类，最后的输出是一个3*3的Tensor，举个例子： 第123行分别是第123张图片的结果，假设第123列分别是猫、狗和猪的分类得分。 可以看出模型认为第123张都更可能是猫。 然后对每一行使用Softmax，这样可以得

>>阅读原文<<