pytorch 中 [..., 0] 的理解

问题描述：[…， 0]的使用

[…, 0] == [:, :, 0]， 三个点号…等于x个冒号:，0 表示取第1轴上的所有数据

import torch
a = torch.rand((6, 13, 12, 8))  # 共有8轴数据
b = a[..., 0]  # 取第1轴上的所有数据 ，数据size=[6,13,12]
c = a[:, :, :, 0]  # 取第1轴上的所有数据 ，数据size=[6,13,12]
print(b.size())
print(c.size())

d = torch.rand((13, 12, 8))
b1 = d[..., 0]  # 取第1轴上的所有数据 ，数据size=[13,12]
c1 = d[:, :, 0]  # 取第1轴上的所有数据 ，数据size=[13,12]
c2 = d[:, 0]  # 取8个轴上的第一行第一列的数据 数据size=[13,8]
print(b1, c2)
print(b1.size(),'\n',c1.size(), c2.size())

解释：

a = torch.rand((6, 13, 12, 8)) 共有8轴数据

b = a[…, 0] 取第1轴上的所有数据 ，数据size=[6,13,12]

c = a[:, :, :, 0] 取第1轴上的所有数据 ，数据size=[6,13,12]

即：b = c

参考资源：[1]https://blog.csdn.net/qq_17547195/article/details/107623716?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7Edefault-1.no_search_link&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7Edefault-1.no_search_link