qq_42178626的博客

拆分与合并合并方法：cat()stack()拆分方法：split()chunk()cat()用法cat([x1,x2,...],dim=Y)把X1,X2,在维度为Y上面，进行合并前提除了合并维度Y，其余维度必须相同实例#cat#cat([x1,x2,...],dim=Y) dim代表在哪一个维度上进行合并,以x1为主#原则：除了dim维度上，其他的维度必须一致res=torch.cat([a,b],dim=0)print(res.shape)

索引与切片#隔行采样#每个图片从0开始到28,每隔两个像素进行采样a[:,:,0:28:2,0:28:2].shape隔行采样#生成4个图片，每个图片有3个通道，像素为28*28a=torch.randn(4,3,28,28)#取前两个图片a[:2].shapeout:torch.Size([2,3,28,28])给定值进行采样#对第一张图片进行采样a.index_select(0,torch.tensor([1,2]))shape维度变换view()