https://soondong2.github.io//posts/Softmax-and-Cross-entropy/
[PyTorch] Softmax