當我運行程式時，A 和 B 矩陣會有所不同
A = np.array([[1, 1, 1], [2, 2, 2]])
B = np.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]])
輸出矩陣 ( C) 應與 matrix 具有相同的維度A。正如標題所說，我試圖將每一行從一個矩陣 ( A) 乘以每一行到另一個矩陣 ( B)，并希望將它們相加。
例如，維度C = (2,3)
C = [[A(0)*B(0) A(1)*B(0)], [A(0)*B(1) A(1)*B(1)],[A(0)*B(1) A(1)*B(1)]]
我想知道是否有一個 numpy 函式可以做到這一點。
uj5u.com熱心網友回復：
使用 numpy 廣播：
C = (A * B[:, None]).sum(axis=1)
輸出：
>>> C
array([[3, 3, 3],
[6, 6, 6],
[9, 9, 9]])
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/448408.html
上一篇：設計模式之享元模式