Python中三维数组与一维内核的卷积

Question

是否可以将一维阵列与三维阵列进行卷积？例如：

A是我想要卷积的8x2x2矩阵。假设A有2x2个子矩阵（A=A7 A6 A5…A0），每个子矩阵是2x2。B是包含标量权重（B0B1 B2 B3 B4）的5x1数组。我要做的是将B数组与数组的第一个维数进行卷积，在本例中为8。我知道numpy.convalve是可用的，但它不支持多维。为了澄清我的例子：Convolution example

Answer 1

使用：

arr_out = np.apply_along_axis(
                lambda x: np.convolve(x, B.flatten(), mode = 'same'), 
                axis = 0, arr = A)