如何使用最大池化从 LSTM 节点收集信息

Question

gru_out = Bidirectional(GRU(hiddenlayer_num, return_sequences=True))(embedded)
#Tensor("concat_v2_8:0", shape=(?, ?, 256), dtype=float32)

我使用 Keras 创建 GRU 模型。我想从 GRU 模型的所有节点向量中收集信息，而不是最后一个节点向量。例如，我需要获取每个向量的最大值，如图像描述，但我不知道如何执行此操作。

Answer 1

可以使用GlobalMaxPooling1D这里描述的：

gru_out = Bidirectional(GRU(hiddenlayer_num, return_sequences=True))(embedded)
max_pooled = GlobalMaxPooling1D(gru_out)