如何从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].

是否可以在不使用任何for循环的情况下访问矩阵A（[[3,4], [3,4]]）的右上子矩阵？天真地使用A[ind_1, ind_2]仅返回元素3和4（所需子矩阵的对角线）。

Answer 1

PyTorch张量支持python切片，因此您可以使用A[:2, 2:]以2x2张量访问前两行和后两列。