a = torch.rand(4,3,28,28) # 4张图片,3通道,高度28,宽度28
a[0].shape # torch.Size([3, 28, 28]) 表示第0张图片的shape是3通道,高度28,宽度28
a[0,0].shape # torch.Size([28, 28]) 表示第0张图片第0个通道的shape
a[0,0,2,4].shape # tensor(0.5649) 表示第0张图片第0个通道第2行第4列的值是0.5649
a[:2].shape # torch.Size([2, 3, 28, 28]) 表示取第0个和第1个图片的shape
a[:2,:1,:,:].shape # torch.Size([2, 2, 28, 28]) 表示前2个图片,第1个通道上的shape
a[:2,1:,:,:].shape # torch.Size([2, 2, 28, 28]) 表示前2个图片,第1个通道之后的shape
a[:2,-1:,:,:].shape # torch.Size([2, 1, 28, 28]) 表示前2个图片,第最后1个通道之后的shape
a[:,:,0:28:2,0:28:2].shape # torch.Size([4, 3, 14, 14]) 每隔2个数据进行采样。
a.index_select(0,torch.tensor([0,2])).shape # torch.Size([2, 3, 28, 28]) 在第0个维度上进行采集第0个图片和第2个图片。
a.index_select(1,torch.tensor([1,2])).shape # torch.Size([4, 2, 28, 28]) 在第1个维度上进行采集第2个通道和第3个通道
a.index_select(2,torch.arange(28)).shape # torch.Size([4, 3, 28, 28]) 在第2个维度上选取所有的行
a[...].shape # torch.Size([4, 3, 28, 28])表示获取所有的维度。
a[0,...].shape # torch.Size([3, 28, 28])
PyTorch教程(三):索引与切片
最新推荐文章于 2024-05-28 14:39:12 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)