Numpy Matrixmultiplikation mit benutzerdefinierten Punktprodukt

Question

Die Standard-Matrix-Multiplikation berechnet wird als

c[i,j] = sum(a[i,k] * b[k,j]) 

Ich versuche, an Stelle des Punktprodukt eine benutzerdefinierte Formel zu verwenden

c[i,j] = sum(a[i,k] == b[k,j]) 

Gibt es ein effizienter zu bekommen Weg, dies in numpiger Weise zu tun?

Answer 1

Sie könnten Rundfunk verwenden:

c = sum(a[...,np.newaxis]*b[np.newaxis,...],axis=1) # == np.dot(a,b) 

c = sum(a[...,np.newaxis]==b[np.newaxis,...],axis=1) 

ich die newaxis in b enthalten nur deutlich machen, wie das Array erweitert wird. Es gibt andere Möglichkeiten, Dimensionen zu Arrays hinzuzufügen (Umformen, Wiederholen, usw.), aber der Effekt ist der gleiche. Erweitern Sie a und b in die gleiche Form, um Element für Element zu multiplizieren (oder ==), und addieren Sie dann auf der richtigen Achse.