... 和：在 Pytorch 张量和 numpy 索引中有什么区别

Question

我正在尝试习惯 Pytorch 索引。 但是我无法理解 tensor[:,-1] （应该打印最后一列）和 tensor[...,-1] 之间的区别，它打印不同的 output （输出2）

import torch
tensor = torch.rand([3,3,3,3])
print('Output1')
print(tensor[:,-1])
print('Output2')
print(tensor[...,-1])

Answer 1

看起来以下索引是等效的

tensor[:, -1] == tensor[:, -1, :, :]
tensor[..., -1] == tensor[:, :, :, -1]