是否有一个相当于MATLAB conv2（h1，h2，A，'same'）的python？

您可能想要类似的东西：

def conv2(v1, v2, m, mode='same'):
    """
    Two-dimensional convolution of matrix m by vectors v1 and v2

    First convolves each column of 'm' with the vector 'v1'
    and then it convolves each row of the result with the vector 'v2'.

    """
    tmp = np.apply_along_axis(np.convolve, 0, m, v1, mode)
    return np.apply_along_axis(np.convolve, 1, tmp, v2, mode)

适用于MATLAB文档中conv2的示例：

A = np.zeros((10, 10))
A[2:8, 2:8] = 1
x = np.arange(A.shape[0])
y = np.arange(A.shape[1])
x, y = np.meshgrid(x, y)

u = [1, 0, -1]
v = [1, 2, 1]

Ch = conv2(u, v, A, 'same')
Cv = conv2(v, u, A, 'same')

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

plt.figure()
ax = plt.gca(projection='3d')
ax.plot_surface(x, y, Ch)

plt.figure()
ax = plt.gca(projection='3d')
ax.plot_surface(x, y, Cv)