python中如何对两个矩阵逐列进行线性卷积

Question

我有两个矩阵，我需要在矩阵A的每一列和矩阵B的对应列之间进行卷积，并将结果存储在矩阵y。你知道吗

例如，A是维数为10 x 5的复矩阵，b也是维数为7 x 5的复矩阵，所以我可以在matlab中这样做：

A = randi(10,5) + 1j*randi(10,5); 
B = randi(7,5) + 1j*randi(7,5); 
for i = 1 : 5 
    y(:,i) = conv(A(:,i),B(:,i));
end

因此，当我在python中构建该步骤时，如下所示：

for i in range(5):
    y[:,i]= np.convolve(A[:, i], B[:, i])

这给了我一个错误

UnboundLocalError: local variable 'y' referenced before assignment

任何人都可以帮助解决这个问题。我还是python的新手！！你知道吗

Answer 1

如上所述，您可以在for循环之前添加这一行

y = np.zeros((10+7-1,5)) 

你们知道，两个向量卷积的结果向量的长度是它们的长度之和-1，因为我用了（10+7-1），这是对任意两个向量的推广。你知道吗

祝你好运