Hi,
What is the expected output? A list of Tensors?
If your Tensors in the list don’t all have the same size, you won’t be able to pack them into a single Tensor.
What might help though is to concatenate all the indices into a 1D Tensor. Then you can do all the indexing in one go.
If you still want the list as output, you can get it by using narrow(0, curr_idx, nb_el) for each element.