簡體   English   中英

numpy矩陣乘法

[英]numpy matrix multiplication

我試圖找出如何在numpy中進行一種標量矩陣乘法。

我有

a = array(((1,2,3),(4,5,6)))
b = array((11,12))

而且我想做

a op b

以導致

array(((1*11,2*11,3*11),(4*12,5*12,6*12))

現在我使用以下表達式

c = a * array((b,b,b))。transpose()

似乎必須有一種更有效的方法來做到這一點

利用廣播

(a.T * b).T

轉置a的替代方法是更改b的形狀以使廣播產生您正在尋找的結果:

a * b[:, np.newaxis]

請注意,將新軸添加到b會給出以下數組:

array([[11],
       [12]])

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM