tan_1999的博客

一看到切片就发蒙，，稍微整理一下。 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[:2,:1].size()) print(a[:2,:1]) 3.a[:2,:,2:] 取第一维度的前两维，