如何使用最大池从 LSTM 节点收集信息
How to use max pooling to gather information from LSTM nodes
gru_out = Bidirectional(GRU(hiddenlayer_num, return_sequences=True))(embedded)
#Tensor("concat_v2_8:0", shape=(?, ?, 256), dtype=float32)
我使用 Keras 创建了一个 GRU model.I 想从 GRU 模型的所有节点向量中收集信息,而不是最后一个节点向量。
例如,我需要获取每个向量的最大值,就像图像描述一样,但我不知道该怎么做。
可以使用此处描述的 GlobalMaxPooling1D
:
gru_out = Bidirectional(GRU(hiddenlayer_num, return_sequences=True))(embedded)
max_pooled = GlobalMaxPooling1D(gru_out)
gru_out = Bidirectional(GRU(hiddenlayer_num, return_sequences=True))(embedded)
#Tensor("concat_v2_8:0", shape=(?, ?, 256), dtype=float32)
我使用 Keras 创建了一个 GRU model.I 想从 GRU 模型的所有节点向量中收集信息,而不是最后一个节点向量。
例如,我需要获取每个向量的最大值,就像图像描述一样,但我不知道该怎么做。
可以使用此处描述的 GlobalMaxPooling1D
:
gru_out = Bidirectional(GRU(hiddenlayer_num, return_sequences=True))(embedded)
max_pooled = GlobalMaxPooling1D(gru_out)