pytorch维度转换

四维特征图

BCHW(B批处理大小，C通道数,H高，W宽)

三维特征图

BNC(N在图像处理中可以理解为H*W，在时序中可以理解为时间序列的长度)

#四维的张量转换成三维

1.x=torch.random(10,3,32,32)

#转换成BNC,

b,c,h,w=x.size()

out=x.view(b,h*w,c)

2.x=torch.random(10,3,32,32)

   x_out=x.permute(0,2,3,1)

#从下标为1的位置开始展平，到起始位置为2结束 

   out=x_out.flatten(start_dim=1,end_dim=2)

   print(out.shape)