多层简单神经网络在PyTorch中的实现

  MNIST数据集是学习各种神经网络的开胃菜。它的任务是识别手写数字（1~9），具体介绍官网http://yann.lecun.com/exdb/mnist/。这次我们用MNIST来小试牛刀。

1. 准备工作

1.1 导包和设置随机种子

import torchvision
import torchvision.datasets as datasets #为了下一步加载数据
import torch
import torch.nn.functional as F

import numpy as np 
import random #设置随机数

USE_CUDA = torch.cuda.is_available() #GPU可用的标志位
random.seed(1)
np.random.seed(1)
torch.manual_seed(1)  

if USE_CUDA:
    torch.cuda.manual_seed(1)

1.2 加载数据和归一化

train_mnist = datasets.MNIST(root = './data', train = True, download = True) #训练集
test_mnist = datasets.MNIST(root = './data', train = False, d