So first let me describe my use case. I have a tensor of shape NxCxseq_len. I have an index tensor of shape NxK (basically K from each batch) where K denotes the indices from the seq_len axis and I need to pick these specific samples. I am currently using a loop to retrieve a sample corresponding to every column in the K axis but I feel there must be an easier way to go about this. Btw the indices are not the same for each sample in the batch.
So my resultant tensor should be of the shape NxCxK