LSTM在MNIST手写数据集上做分类(代码中尺寸变换细节)
RNN和LSTM学了有一段时间了,主要都是看博客了解原理,最近在研究SLSTM,在对SLSTM进行实现的时候遇到了困难,想说先比较一下二者的理论实现,这才发现自己对于LSTM内部的输入输出格式、输出细节等等都不是非常清楚,借此机会梳理一下,供后续研究使用。
下面代码来自github的mnist_master项目,非常基础且工整。也是我学习RNN、CNN的入手代码示例,感谢!
from ten...
原创
2018-11-15 16:56:18 ·
1410 阅读 ·
0 评论