假设我有两个不同行数的二维数组:
我希望能够沿着零轴将这两个数组进行卷积,输出大小与数组1相同,类似于使用mode='same'时,scipy的convolve方法输出的内容。
因此,两个数组的第一列进行卷积,以此类推,两个数组的第二列进行卷积等等。我希望以一种对于更大的数组(超过100,000行)来说计算可行的方式来完成这个任务,所以如果可能的话,我想避免使用for循环。是否有numpy/scipy或其他库可以执行此操作?
期望的输出结果应如下所示:
数组一:
1 2 3
4 5 6
7 8 9
数组二:
10 11 12
13 14 15
我希望能够沿着零轴将这两个数组进行卷积,输出大小与数组1相同,类似于使用mode='same'时,scipy的convolve方法输出的内容。
因此,两个数组的第一列进行卷积,以此类推,两个数组的第二列进行卷积等等。我希望以一种对于更大的数组(超过100,000行)来说计算可行的方式来完成这个任务,所以如果可能的话,我想避免使用for循环。是否有numpy/scipy或其他库可以执行此操作?
期望的输出结果应如下所示:
10 22 36
53 83 117
122 158 198
输出的形状与第一个输入数组相同。每一列都是数组1和数组2中相应列卷积得到的输出。