多輸入計算矩陣產品

Question

我正試圖計算由b = A * x給出的變換。 A是（3,4）矩陣。如果x是一個（4,1）向量，則結果是b（3,1）。

相反，對於x我有一堆連接成一個矩陣向量的和我試圖評估變換x的每個值。所以x是（20,4）。如何在numpy中廣播這個數據，以便得到b（20,3）的20個結果值？

我可以遍歷每個輸入和計算輸出，但它的感覺就像必有用廣播的更好的方法。

例如，

A = [[1,0,0,0], 
[2,0,0,0], 
[3,0,0,0]] 

如果x是：

x = [[1,1,1,1], 
[2,2,2,2]] 

b = [[1,2,3], 
[2,4,6]] 

x的每一行乘以A和結果存儲爲b中的行。

Answer 1

numpy的點

import numpy as np 

A = np.random.normal(size=(3,4)) 

x = np.random.normal(size=(4,20)) 

y = np.dot(A,x) 

print y.shape 

結果：（3,20）

當然，如果你想（20,3），你可以使用np.transpose（）