Как я могу получить доступ к подматрице, указанной индексными векторами строк и столбцов, из матрицы (2D-тензор) в pytorch?

Question

Пожалуйста, рассмотрите матрицу A= torch.tensor([[1,2,3,4], [1,2,3,4], [1,2,3,4], [1,2,3,4]] и векторы индексов ind_1= [0,1] и ind_2= [2,3].

Возможно ли получить доступ к верхней правой подматрице матрицы A ([[3,4], [3,4]]) без использования есть петли? Наивно используя A[ind_1, ind_2] просто возвращает элементы 3 и 4 (диагональ требуемой подматрицы).

Answer 1

Тензоры PyTorch поддерживают нарезку python, поэтому вы можете использовать A[:2, 2:] для доступа к первым двум строкам и двум последним столбцам как тензор 2x2.