Предположим, у меня есть 2d Tensor:
import tensorflow as tf
import numpy as np
matrix = np.ones((10000,10000))
matrix_tf = tf.convert_to_tensor(matrix)
Я хочу выбрать строки и столбцы на основе списка признаков:
ind = [1,3,5,7]
Это означает, что я хочу выбрать строку и столбец 1, строка и столбец 3, строка и столбец 5, строка и столбец 7.
Я пробовал много способов, но лучше всего работает:
temp = tf.gather(matrix,ind)
temp2 = tf.gather(temp,ind, axis = 1)
Но это не так не выглядит хорошо. Как я могу сделать это более эффективно?