關於pytorch在使用torch.nn.functional.cross_entropy的注意點

在使用pytorch實現textcnn進行文本分類時，參考了別人的代碼(keras)，在使用torch.nn.functional.cross_entropy糾結了好久，最後發現的下面的文章：html https://www.cnblogs.com/marsggbo/p/10401215.html網絡 總結了如下的注意點：編碼 在網絡的forward最後輸出時不用接softmax，直接全鏈接輸出n

>>阅读原文<<