NumPy nhân ma trận với tùy chỉnh dot sản phẩm

Question

Các nhân ma trận mặc định được tính như

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

Tôi đang cố gắng sử dụng một công thức tùy chỉnh thay vì các sản phẩm chấm để có được

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

Có một hiệu quả cách để làm điều này trong numpy?

Answer 1

Bạn có thể sử dụng phát sóng:

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

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

tôi bao gồm các newaxis trong b chỉ làm cho nó rõ ràng như thế nào mảng được mở rộng. Có nhiều cách khác để thêm kích thước vào mảng (định hình lại, lặp lại, v.v.), nhưng hiệu quả là như nhau. Mở rộng a và b cho cùng một hình dạng để làm phần tử theo yếu tố nhân (hoặc ==), sau đó tổng hợp trên trục chính xác.