如何在Python上将一个矩阵的每一行乘以另一个矩阵的每一行？

Question

当我运行程序时，A 和 B 矩阵将不同

A = np.array([[1, 1, 1], [2, 2, 2]])
B = np.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]])

输出矩阵 (C) 应与矩阵 A 具有相同的维度。 正如标题所述，我试图将一个矩阵 (A) 的每一行乘以另一个矩阵 (B) 的每一行，并希望对它们求和。

例如， C = (2,3) 的维度

C = [[A(0)*B(0) + A(1)*B(0)], [A(0)*B(1) + A(1)*B(1)],[A(0)*B(1) + A(1)*B(1)]]

我想知道是否有 numpy 函数可以做到这一点。

Answer 1

使用 numpy 广播：

C = (A * B[:, None]).sum(axis=1)

输出：

>>> C
array([[3, 3, 3],
       [6, 6, 6],
       [9, 9, 9]])