Using an ellipsis for the middle dimensions of a PyTorch tensor

Question

Suppose I have a torch.Tensor t of shape (10, 20, 20, 10). I want to index along the first and last dimensions only.

In this case, since I k