紧张雨

文章
9
资源
0
加入时间
3年0月28天

pytorch之手写数字识别

记一下学习pytorch一次程序练习,主要功能是实现构造三层神经网络识别MNIST数据集,最后的识别结果:本文代码如下:import torchimport numpy as npimport matplotlib.pyplot as pltfrom torch import nnfrom torch import optimimport osfrom torch.autograd import Variablefrom torch.utils.data import DataLoader