在keras中预测时索引越界

Question

我正在尝试在keras中定制一个特殊的注意层。但是我很困惑为什么这个错误总是在尝试了很多方法之后发生。在

Traceback (most recent call last):
 File "D:/Users/LawLi/PyCharmProjects/fixed_talentDNA/adx.py", line 52, in <module>
   print(model.predict([tensor1, tensor2, indices]))  # (bs1, sl1, sl2)
 File "D:\Users\LawLi\Anaconda3\lib\site-packages\keras\engine\training.py", line 1172, in predict
   steps=steps)
 File "D:\Users\LawLi\Anaconda3\lib\site-packages\keras\engine\training_arrays.py", line 293, in predict_loop
   ins_batch = slice_arrays(ins, batch_ids)
 File "D:\Users\LawLi\Anaconda3\lib\site-packages\keras\utils\generic_utils.py", line 507, in slice_arrays
   return [None if x is None else x[start] for x in arrays]
 File "D:\Users\LawLi\Anaconda3\lib\site-packages\keras\utils\generic_utils.py", line 507, in <listcomp>
   return [None if x is None else x[start] for x in arrays]
IndexError: index 4 is out of bounds for axis 0 with size 4

这是我关于自定义层的测试代码，代码包含预测。以及它可以直接运行。在

^{pr2}$

谢谢你的帮助。在

Answer 1

预测函数无法接受轴0大小不同的输入。所以我尝试了另一个名为predict_on_batch()的函数