Loading

python中的...操作

假如输入数据是4维的

a = torch.randn(3, 4, 5, 6)
y = a[..., 0] # 取第四个维度的第0个
上面操作等同于
y = a[:, :, :, 0]
posted @ 2021-06-11 11:27  Guang'Jun  阅读(57)  评论(0)    收藏  举报