tensorflow - Tensorflow：向 LSTM 添加前馈

Question

如何使用 Tensorflow 在标准 LSTM 网络之前添加额外的前馈层？

标签： tensorflowlstmfeed-forward

Answer 1

n_timesteps = 15
n_dim = 5

model = models.Sequential()
model.add(layers.TimeDistributed(layers.Dense(10, activation='relu'), input_shape=(n_timesteps, n_dim)))
model.add(layers.LSTM(25))
model.add(layers.Dense(2, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adam')