一看到切片就发蒙,,稍微整理一下。 a=torch.rand(4,3,5) 1.a[:2] 取第一维度的前两维 print(a) print(a[:2].size()) print(a[:2]) 2.a[:2,:1] (写法等同于 a[:2,:1,:]) 取第一个维度的前两维,第二个维度的前一维,第三个维度全部取到。 a=torch.rand(4,3,5) print(a) print(a