每天五分钟深度学习PyTorch:Tensor张量的索引和切片 |
您所在的位置:网站首页 › python中张量切片 › 每天五分钟深度学习PyTorch:Tensor张量的索引和切片 |
本文重点
有时候当我们拥有一个Tensor张量的时候,我们可能需要获取它某一维度的信息,那么此时我们就需要索引和切片的技术,它们可以帮助我们解决这些问题。 切片操作a是四维的,然后默认是从第一维开始取,逗号表示取不同的维度 a[:2]表示第一维取0,1,后面三维取所有 a[:2,:2]表示第一维取0,1,第二维取0,1,后面两维取所有 前两维取所有,第三维和第四维间隔取样,0到28,每两维取样一次 0:28:2等于::2 索引下标通过下标进行索引 index_select 通过特殊的索引来获取数据index_select,这个这样来理解,第一个参数表示a的第几维度,第二个参数表示获取该维度的哪部分。 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |