Matrix Index Selection

1

Hi all,

We have a problem that how we can select the rows from the matrix A using torch.index_select?
The matrix A.size() = (14, 10), the indices B.size() = (196,).
B = tensor([0, 0, …, 13])

Thanks.

C = torch.index_select(A, 0, B) can work well.

2

C = torch.index_select(A, 0, B) can work well.