1. 示例1
import torch
a = torch.rand((17, 24, 8))
b = a[..., 0]
c = a[:, :, 0]
print(b == c)
结果为True
2. 示例2
import torch
a = torch.rand((64, 17, 24, 8))
b = a[..., 0]
c = a[:, :, :, 0]
print(b == c)
结果为True
3. 结论
可以看出[…, 0]相当于[:, :, … :, 0]