深度学习01-Pytorch实现mnist手写数字识别

• 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
• 🍖 原作者：K同学啊

1.我的环境：

2.代码：

3.个人总结：

1.理解了卷积运算的过程，知道输入一个尺寸后，如何经过卷积得到输出的尺寸。

2.对于torch中的squeeze 函数，通常用于删减维度的·操作，这里主要是matplot要查看的图像应该要是三维的或者二维的，之前是张量的形式。

3.iter()函数和next函数一般是一起用的，iter（）将数据转换为可迭代对象，next进行一个批次的查询。

4.torch.argmax 函数一般是返回最大值所在的索引，一般主要用在计算准确率的时候。

5.网络架构那里flatten函数也可以换成view函数 x = x.view(x.size(0), -1)。

​