gru_out = Bidirectional(GRU(hiddenlayer_num, return_sequences=True))(embedded)#Tensor("concat_v2_8:0", shape=(?, ?, 256), dtype=float32)
我使用Keras创建了一个GRU模型。我希望从GRU模型的所有节点向量中收集信息,而不是仅从最后一个节点向量中获取。例如,我需要获取每个向量的最大值,就像图像描述那样,但我不知道该如何操作。
回答:
可以使用这里描述的GlobalMaxPooling1D
来实现:
gru_out = Bidirectional(GRU(hiddenlayer_num, return_sequences=True))(embedded)max_pooled = GlobalMaxPooling1D(gru_out)