將二維矩陣的每一行與向量進行卷積

Question

我有一個 1000000x1000 (MxN) 矩陣A 。 我有另一個大小為 L 的向量b 。我需要將二維矩陣A 的每一行與向量b進行卷積。 我怎么能在python中做到這一點？

我試過了， C = np.convolve(A, b)

但我收到一個錯誤說

in convolve return multiarray.correlate(a, v[::-1], mode)
ValueError: object too deep for desired array

Answer 1

嘗試scipy 的 convolve2d

C = scipy.signal.convolve2d(A, b)

只需確保len(b.shape) == 2 （意味着它是一個二維數組，一維大小為 1）

這將起作用，因為b過濾器將滑過A每一行，在 C 中產生一個新行，然后跨到下一行，執行相同操作，創建另一行，依此類推。