I have a tensor of size torch.Size([3307, 1, 3]) containing values and another tensor of size torch.Size([3307, 1, 1]) containing indices of values I want to select. I am unable