本文重点
有时候当我们拥有一个Tensor张量的时候,我们可能需要获取它某一维度的信息,那么此时我们就需要索引和切片的技术,它们可以帮助我们解决这些问题。
切片操作
a是四维的,然后默认是从第一维开始取,逗号表示取不同的维度
a[:2]表示第一维取0,1,后面三维取所有
a[:2,:2]表示第一维取0,1,第二维取0,1,后面两维取所有
前两维取所有,第三维和第四维间隔取样,0到28,每两维取样一次
0:28:2等于::2
索引下标
通过下标进行索引
index_select
通过特殊的索引来获取数据index_select,这个这样来理解,第一个参数表示a的第几维度,第二个参数表示获取该维度的哪部分。