The indexing array is jagged, how can I make it?
import torch
a = torch.randn(25, 300)
i = [[0,0],[1,1,1],[2,2,2,2]]
j = [[12,10],[13,24,26], [8,9,10,11]]
a[i,j] #error <-expected sequence of length 2 at dim 1 (got 3)