numpy 矩陣向量乘法

| | | | | | | | | |

當我將兩個大小為 (nxn)*(nx 1) 的 numpy 數組相乘時,我得到一個大小為 (nxn) 的矩陣。遵循正常的矩陣乘法規則,需要一個 (nx 1) 向量,但我根本無法在 Python 的 Numpy 模塊中找到有關此操作的任何信息。

問題是我不想要手動實現它以保持程序的速度。

示例代碼如下所示:

a = np.array([[ 5, 1 ,3], [ 1, 1 ,1], [ 1, 2 ,1]]) b = np.array([1, 2, 3]) print a*b >> [[5 2 9] [1 2 3] [1 4 3]] 

我想要的是:

print a*b >> ; [16 6 8]