重新排序numpy 4D数组

Question

我在理解如何管理和修改numpy矩阵时遇到了麻烦。我发现很难“描绘”脑海中的矩阵。

我有一个（4x2x1x1）矩阵，我想将其制成（1x2x1x4）矩阵，这样我就可以将矩阵乘法与另一个形状为（3x2x1x1）的矩阵相乘。

谢谢！

Answer 1

如果您的矩阵称为矩阵，则matrix.shape =（1,2,1,4）（如上述示例所示）可以解决问题。 NumPy会自动通知您新形状是否“超出范围”，如果不是，则会自动正确地对数据重新排序。

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