重新排序 numpy 4D 数组

Question

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

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

提前致谢！

Answer 1

如果你的矩阵被称为矩阵，matrix.shape = (1,2,1,4)（如我上面的例子）就可以了。NumPy 会自动注意到您的新形状是否“越界”，如果不是，则会自动正确重新排序数据。编辑：您还可以使用 newMatrix = numpy.reshape(matrix, (1,2,1,4)) 创建一个新矩阵作为您的第一个矩阵的重塑。