Numpy兩個矩陣，行的成對點積

Question

我們目前正在研究一個python項目，並且由於性能限制而必須進行很多矢量化處理。我們結束了以下計算：我們有兩個numpy數組，形狀爲(20,6)，並且想要計算行的成對點積，即我們應該在最後獲得(20,1)矩陣，其中每一行都是由相應向量獲得的標量點乘法。

Answer 1

可以明智乘兩個數組元素，然後通過行做總和，那麼你有一個數組，其中每個元素都是從原來的兩個陣列行的點積：

a = np.array([[1,2], [3,4]]) 
b = np.array([[3,4], [2,1]]) 

(a * b).sum(axis=1) 
# array([11, 10])