x = Input(shape=(timesteps, input_dim,))
# LSTM encoding
h = LSTM(2048)(x)
这是我从互联网上下载的文件中的几行代码。我认为h
适用于具有2048个单位的单层LSTM层。如何使它成为多层,即2个隐藏层。
答案 0 :(得分:2)
只需添加另一个图层(让我们称之为g
)!但是,由于我们要转移到另一个LSTM
图层,因此我们必须将return_sequences
关键字参数添加到第一层,以便我们可以将正确的输入形状添加到第二层。 / p>
x = Input(shape=(timesteps, input_dim,))
# LSTM encoding
h = LSTM(2048, return_sequences=true)(x)
g = LSTM(10)(h)