RNN:在Keras框架的LSTM层中return_sequences的用途是什么

时间:2018-07-17 08:37:39

标签: machine-learning keras lstm recurrent-neural-network rnn

我正在RNN工作。我有一些网站的以下代码行。 如果您观察到第二层没有“ returnSequence”参数。

我假设返回序列是强制性的,因为它应该返回序列。 您能告诉我为什么未定义吗?

第一层LSTM:

regressor.add(LSTM(units = 30, return_sequences = True))

第二层LSTM:

regressor.add(LSTM(units = 30))

1 个答案:

答案 0 :(得分:1)

return_sequences参数设置为False(默认值)时,网络将仅输出h n ,即最后一个时间步的隐藏状态。否则,网络将输出隐藏状态的完整序列[h 1 ,h 2 ,...,h n ]。层的内部方程式不变。请参阅documentation