Pytorch中的tensor[...,0]

参考博客：
PyTorch中[…, 0]的使用
pytorch中[…, 0]的用法

>>> a= torch.rand(1,2,2,2,2)
>>> a
tensor([[[[[0.2528, 0.9359],
           [0.1876, 0.4189]],

          [[0.5546, 0.1931],
           [0.1389, 0.7482]]],


         [[[0.6169, 0.2977],
           [0.1109, 0.5175]],

          [[0.4135, 0.3874],
           [0.9714, 0.8702]]]]])
>>> a[...,0]
tensor([[[[0.2528, 0.1876],
          [0.5546, 0.1389]],

         [[0.6169, 0.1109],
          [0.4135, 0.9714]]]])
>>> a[...,1]
tensor([[[[0.9359, 0.4189],
          [0.1931, 0.7482]],

         [[0.2977, 0.5175],
          [0.3874, 0.8702]]]])