nn.CrossEntropyLoss （內置了Softmax） 不要再用softmax

CrossEntropyLoss  （內置了Softmax） 不要再用softmax

 

計算公式如下：x爲一個num_class(類別數量)維的向量，class爲樣本的類別。

參考

1. https://blog.csdn.net/geter_CS/article/details/84857220

2. https://blog.csdn.net/Tianweidadada/article/details/82630735