Numpy zwei Matrizen, paarweise Skalarprodukt Reihen

Question

Wir sind derzeit auf einem Python-Projekt arbeiten und haben eine Menge wegen Leistungsbeschränkungen vektorisieren. Wir am Ende mit der folgenden Berechnung: Wir haben zwei numpy Arrays von Form (20,6) und wollen die paarweise Skalarprodukt der Zeilen berechnen, dh sollten wir eine (20,1) Matrix am Ende erhalten, wobei jede Zeile die skalare durch den jeweiligen Vektor erhalten wird, Punktmultiplikation

Answer 1

Sie können die beiden Arrays elementweise multiplizieren und dann durch Reihen Summe tun, und dann haben Sie ein Array, wobei jedes Element aus Reihen der beiden ursprünglichen Arrays ein Punktprodukt ist:

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

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