我有一个pytorch张量具有任意数量的维度:…X,Y,Z
我想有这样一个函数，我给出一个数字C，然后我得到…，C,Y,Z
my_matrix = [:,:,C,:,:]
但是我不知道在C之前有多少尾随维度，我看到了使用切片元组的答案，但似乎可以让它工作。pytorch/numpy中任意可变维数的部分切片
我认为ellipsis会完成这项工作:
t = torch.randn(2, 3, 6, 5, 9, 3)
t[..., 4, :, :]
u = torch.randn(11, 4, 2, 7)
u[..., 2, :, :].shape