sequence classification example 正好符合您的要求。
唯一的区别是它只使用了一个 LSTM 层。您可以通过更改轻松更改此网络以使用多个层:
LSTM_function = LSTMP_component_with_self_stabilization(
embedding_function.output, LSTM_dim, cell_dim)[0]
到:
num_layers = 2 # for example
encoder_output = embedding_function.output
for i in range(0, num_layers):
encoder_output = LSTMP_component_with_self_stabilization(encoder_output.output, LSTM_dim, cell_dim)
但是,使用新的图层库会更好地为您服务。然后你可以简单地这样做:
encoder_output = Stabilizer()(input_sequence)
for i in range(0, num_layers):
encoder_output = Recurrence(LSTM(hidden_dim)) (encoder_output.output)
然后,要获得要放入密集输出层的最终输出,您可以首先执行以下操作:
final_output = sequence.last(encoder_output)
然后
z = Dense(vocab_dim) (final_output)