pytorch切片与索引

import torch

import numpy as np

切片与索引

a = torch.randint(1,10,[4,3,5,5]) #4张图片，3个通道，长和宽都是28个pixel

a[0].shape #第一张图片的尺寸 0表示第一张

torch.Size([3, 5, 5])

a[1,2].shape #第二张图片，第三个通道的尺寸

torch.Size([5, 5])

a[:,1]