数组维度重塑

Question

我有一个维度为 (300, 2, 17, 80) 的数组，它是 2 张维度 (17,80) 图像的 300 个样本。 我想重塑阵列以馈送我的 CNN，例如 (300, 17, 80, 2)。 我怎样才能做到这一点 ？ 感谢支持。

Answer 1

您可以为此使用 np.moveaxis。例如，

import numpy as np
M = np.zeros((300,2,17,80))
print(M.shape)
M = np.moveaxis(M, 1, -1)
print(M.shape)

np.moveaxis(M, 1, -1) 调用使用轴索引=-1 将轴索引=1 的第二个轴移动到最后一个位置。