使用PyTorch训练MNIST数据集上的卷积神经网络