https://zoshs2.github.io/posts/MnistSoftmax/
[PyTorch] MNIST Dataset - Softmax Regression Model - Pale Blue Dot