I am a little confused about the advanced indexing in a 4d-tensor,
Take an example, suppose I have a 4d-tensor
x = torch.randn(10, 3, 5, 5) (10 RGB images in a mini-batch, the size is 5*5).
And I would like to sample 3 points on each 5*5 image in this mini-batch, the axis of the points are stored in two long tensors, which are
10 * 3 tensors. Can I index these points without using any loop like
[x[i, :, a[i], b[i]] for i in range(10)]?