当前位置:网站首页>举例说明tf中LSTMCell的cell、num_unit是什么意思

举例说明tf中LSTMCell的cell、num_unit是什么意思

2022-06-09 03:51:00 陈壮实的编程生活

1. 对于cell的解释

在这里插入图片描述
LSTM一般都会一连串的小团团,这一个个小团团就表示一个cell。如上图所示。

2. 对num_units参数的解释

在这里插入图片描述
上图中,我们很多步计算中都有用到W, 其实num_units就表示线性层的隐藏节点个数,也即输出的维度。
例如:
num_units = 128, 就表示我们输出的维度是[128, 1](也有可能是[1, 128],看具体需要,两者只差一个转置,没有太大的本质区别)。
我们的输入xt的维度为[28, 1], 那么 [ h t − 1 , x t ] = [ 128 + 28 , 1 ] = [ 156 , 1 ] [h_{t-1},x_t]=[128+28, 1]=[156,1] [ht1,xt]=[128+28,1]=[156,1],所以我们的W的维度为[128, 156]

总之,参数num_units = 128,就表示每一个cell的输出维度为[128, 1] 或 [1, 128]

原网站

版权声明
本文为[陈壮实的编程生活]所创,转载请带上原文链接,感谢
https://blog.csdn.net/qq_41915623/article/details/125174110

随机推荐