每天五分钟深度学习PyTorch:Tensor张量的索引和切片

您所在的位置:网站首页 python中张量切片 每天五分钟深度学习PyTorch:Tensor张量的索引和切片

每天五分钟深度学习PyTorch:Tensor张量的索引和切片

2024-07-08 10:41| 来源: 网络整理| 查看: 265

本文重点

有时候当我们拥有一个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