pytorch实现cnn手写数字集mnist识别

主要分五步：
1.数据块的加载处理
2.模型的建立
3.训练模型
4.预测测试模型
5.保存模型

参考：https://zhuanlan.zhihu.com/p/45500657 （写的不错）
另补小知识：标量，向量，矩阵它们三个也是张量，标量是零维的张量，向量是一维的张量，矩阵是二维的张量 https://zhuanlan.zhihu.com/p/45497796

卷积操作&池化操作的维度变化公式: width_of_output = (width_of_input - filter_size + 2*padding)/stride + 1

import torch
import torch.nn as nn
from torch.autograd import Variable
import torch.utils.data as Data
import torchvision
import time

# import matplotl