今天看程序看到了LSTM,原理都懂,但是想看看参数的个数是怎么算出来的,先简单介绍下我的网络前两层:
(1)第一层是一个 embedding 层,输出是 100 维的。
(2)第二层是一个 LSTM 层,输出是 512 维的。
LSTM这一层的参数个数是 1255424 个,这一看还有整有零的,好奇怎么算出来的,于是百度搜了下,得到这么一个公式:
设 LSTM 输入维度为 x_dim, 输出维度为 y_dim,那么参数个数 n 为:
n = 4 * ((x_dim + y_dim) * y_dim + y_dim)
对应我的网络结构就是:n = 4 * ((100 + 512) * 512 + 512) = 1255424
LSTM 的原理以及公式的又来可以参考这个链接https://www.cnblogs.com/wushaogui/p/9176617.html