I have an index tensor of size [N], that contains indices from 0 to 200
I have a tensor of size [200, N, 3]
I want to somehow do torch.index_select(tensor, 0, index) and get a tensor of size [1, N, 3]. Basically, it needs to select the index in a batch friendly manner etc. How do i do this?