Keras 中 LSTM 中的多层隐藏层

Question

x = Input(shape=(timesteps, input_dim,))
# LSTM encoding
h = LSTM(2048)(x)

这是我从互联网下载的文件中的几行代码。我认为h适用于具有 2048 个单位的单层 LSTM 层。它如何使它成为多层，即 2 个隐藏层。

Answer 1

只需添加另一层(我们称之为g(！但是由于我们要传递到另一个LSTM层，因此我们必须向第一层添加return_sequences关键字参数，以便我们可以为第二层获得正确的输入形状。

x = Input(shape=(timesteps, input_dim,))
# LSTM encoding
h = LSTM(2048, return_sequences=true)(x)
g = LSTM(10)(h)