将二维矩阵的每一行与向量进行卷积

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 中产生一个新行，然后跨到下一行，执行相同操作，创建另一行，依此类推。