qq_45524140的博客

import numpy as npimport torchfrom torchvision import transformsfrom torchvision import datasetsfrom torch.utils.data import DataLoaderimport torch.nn.functional as Fimport torch.optim as optim#MNIST数据集中包含59968张数字图片，大小为（28 * 28），通道数为1 59968 = 64 *