小知识点【python rollaxis()函数。TypeError: Invalid shape (3, 32, 32) for image data】

项目场景：pytorch生成图片

描述：网络输出的数组转化成可视的图片，但是维度不匹配。

---

问题描述

报错：TypeError: Invalid shape (3, 32, 32) for image data

---

原理分析：

提示：下面展示一些 代码片。

rollaxis(a, axis, start = 0)

参数：
a 输入的数组
axis:需要改变的轴，编码从0开始
start:要把axis轴放到哪个轴前，编码从0开始，取值在最后多一位（有n+1个位置可以放）

比如三维数组：33232，想变成32323
axis:012,要变换的是0轴
start: 0123,要放在最后一位，也就是编号3之前，写start = 3

rollaxis(a, axis=0, start = 3)
img = np.rollaxis(AAE_x_hat[i], 0, 3)

---

参考链接

感谢：https://www.cnblogs.com/yqs-0705/p/10174586.html