我重用了用 Keras 编写的逐字注意模型。
我有两个样本输入，input1大小：[批量大小*21*100]，input2大小：[批量大小*2]。基本上，input1是一批句子对，input2是一个token位置列表，每一行包含第二个句子中start token和end token的位置。
对于 input1 中的每一行，在逐字注意操作之后，我得到了一个注意向量列表（大小：11*150），现在我想根据 input2 中的相应行r[]从中提取特定向量，我尝试了类似，但得到错误：rr[input[0]]
TypeError：列表索引必须是整数，而不是 TensorVariable。
谷歌搜索了几天，尝试了很多方法，但没有任何效果。有人有更好的主意吗？谢谢！