【Kaggle-MNIST之路】两层的神经网络Pytorch（四行代码的模型）