重新排序numpy 4D阵列

Question

我很难理解如何管理和修改numpy矩阵。我发现很难“描绘”我头脑中的矩阵。

我有一个（4x2x1x1）矩阵，我想做成一个（1x2x1x4）矩阵，这样我就可以将矩阵乘法应用于另一个具有形状（3x2x1x1）的矩阵。

提前致谢！

Answer 1

如果您的矩阵称为矩阵，则matrix.shape =（1,2,1,4）（如上例所示）可以解决问题。如果您的新形状“超出界限”，NumPy会自动注意到，如果不是，则自动重新排序数据。

编辑：您还可以使用newMatrix = numpy.reshape（矩阵，（1,2,1,4））创建一个新矩阵作为第一个矩阵的重塑。