专栏推荐
本文重点
通过下标进行索引
切片操作
a是四维的,然后默认是从第一维开始取,所以a[:2]表示第一维按照:2取,后面三维取所有
切片,隔行取样
前两维取所有,第三维间隔取样,0到28,每两维取样一次
0:28:2等于::2
通过特殊的索引来获取数据index_select
我们把16,3,28,28看作是16张大小为28的三通道图片,如果我们要想只获得第1张和第三张图片,我们可以这样操作a.index_select(0,torch.tensor([0,2]))
第一个参数0表示操作数据a的第一个维度,也就是第几张图片,第二个参数表示获取第一维度的第一张还有第三张图片,注意第二个参数必须为tensor类型
将数据展开,然后通过索引获取数据
虽然a是2*2的,但是take方法,会先将a展开,然后从0开始设置索引,然后获取a中索引为0和2的
a.ge(0.5)表示a中大于0.5的为1,小于0.5的为0.然后以次为索引获取到a中对应的元素