Except torch.gather does require the same number of dimensions. I assume that if I take the index tensor to size [1,5,8,1,1,1], it’ll just give me the first pixel of each image, so how do I leave the images intact when taking the parts?