Модель внимания Кераса: как динамически выделять векторы из вектора внимания
Я повторяю слово за словом модель внимания, написанную в Керасе.
У меня есть два примера ввода, input1 размер: [размер партии *21*100], input2 Размер: [размер партии *2]. По сути, input1 - это пара пар предложений, input2 - список позиций токена, каждая строка содержит позицию начального токена и конечного токена во втором предложении.
Для каждой строки в input1, После пословной операции внимания, я получил список вектора внимания r[] (размер: 11*150), теперь я хочу извлечь конкретный вектор из r в соответствии с соответствующей строкой в input2, я попытался что-то вроде r[input[0]], но получил ошибку:
Ошибка типа: индексы списка должны быть целыми числами, а не TensorVariable.
Много гуглил несколько дней, пробовал много способов но ничего не работает. У кого-нибудь есть идея получше? Спасибо!