У меня есть тензор размера (n,) целых чисел и 2D тензор размера = (n, k).Для каждой «строки» с индексом i в двумерном тензоре я хочу индексировать ее номером в индексе i одномерного тензора.
Идея состоит в том, чтобы делать это с внутренними функциями Keras только так, как язапуск его на графическом процессоре.
С обработкой списка я бы сделал это так:
array = [1,2,0,0]
matrix = np.array([[1,2,3],[2,3,4],[2,3,1],[3,2,1]])
print([matrix[i, num] for i, num in enumerate(array)])