Keras中LSTM中的多层隐藏层

时间:2017-08-17 04:06:53

标签: python deep-learning keras keras-layer

x = Input(shape=(timesteps, input_dim,))

# LSTM encoding
h = LSTM(2048)(x)

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

1 个答案:

答案 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)