深度学习入门之PyTorch学习笔记＃4
RNN循环神经网络识别MNIST手写数字集，把每张图片看做长为28，特征维度为28的序列，虽然很好理解，但是没想到能有这种操作。。。
现放训练效果图：
里面感觉最需要注意的就是维度的变换，输入数据的维度是batch,1,28,28，而RNN输入应该是28,batch,28，所以首先需要squeeze变成batch,28,28，然后可以用permute进行维度顺序的改变，或者在RNN中使用batch_first=True参数
- 发表于:
- 原文链接：http://kuaibao.qq.com/s/20180113G08TCM00?refer=cp_1026
- 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号（企鹅号）传播渠道之一，根据《腾讯内容开放平台服务协议》转载发布内容。
- 如有侵权，请联系 cloudcommunity@tencent.com 删除。
相关快讯
扫码
添加站长 进交流群
领取专属 10元无门槛券
私享最新 技术干货
腾讯云开发者
