python - numpy中的3-D numpy矩阵乘法得到一维数组

Question

我有 2 个这种形式的 3-D 矩阵：

a = np.random.randint(0, 100, size=(3, 4, 3))

b = np.random.normal(0, 1, size=(3,3, 4))

我想将这两个矩阵相乘，这样我得到一个这种形式的矩阵：

a_b = np.random.normal(0,1,size = (3,1))

我无论如何都可以改变矩阵的形状b来获得a_b形状。例如，我可以b 2X2 matrix用任何其他方式制作或改变矩阵的形状。

有人可以指导我吗？

谢谢！

标签： pythonnumpymatrix