两种变换维度的方法

x.permute（） 更换维度顺序
x.size（）提取维度
x.view（）转换维度

bchw  转换为BNC为（b，hw，c）

1.法一

B,C,H,W = x.size（）赋予单独变量保存

x_reshape = x.view(B,H*W,C)

print(x_reshape.reshape)

2.法二

x = torch.randn（10，3，32，32） 随机赋予一个张量 bchw
x_out = x.permute（0，2，3，1） BNC  变换位置 b-0 hw-2,3  c-1

out=x.out.flatten（start_dim=1，end_dim=2）
print（out.shape）