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:]取第一维度的前两维，