Minor Review: Exercise - Pytorch Softmax Regression