我是Ingoo Lee，正在研究生物信息学。
我想从LSTM批量查询，但我不知道怎么做
让我们假设keras中的LSTM层接受嵌入序列并返回每个步骤的隐藏状态序列。
lstm_result = LSTM(hidden_dim ,return_sequences=True)
因此其结果的形状将是这样
(size_of_batch, length_of_sequence, hidden_dim)
这是输入，它是根据序列中每个类型步骤的索引
input_target = Input(shape=(length_of_sequence))
因此，它的形状就是这样
(size_of_batch, length_of_sequence)
我想从LSTM结果中收集信息。
这是我的输入示例，经过训练的LSTM的形状与输入相同
[[3,5]]
它将收集LSTM的第三和第五个结果并将其连接
因此，我们假设隐藏尺寸为64，那么收集形状将为
(1,2,64)
正如我所说..我不知道如何收集和查询这份工作。
也许我可以将输入用作索引的一键式。
感谢您阅读我的提问！