Свертка только по одной оси

У меня есть два двумерных массива с одинаковыми размерами первой оси. В python я хотел бы свернуть две матрицы только по второй оси. Я хотел бы получить C ниже, не вычисляя также свертку по первой оси.

import numpy as np
import scipy.signal as sg

M, N, P = 4, 10, 20
A = np.random.randn(M, N)
B = np.random.randn(M, P)

C = sg.convolve(A, B, 'full')[(2*M-1)/2]

Есть ли быстрый способ?