¿Qué funciones se pueden utilizar para la convolución 3D de Python?

Escribí una función de convolución, la entrada y el núcleo de convolución dim=2 son iguales (ambos son 3), intenta agregar un bucle for adicional para convertirlo en convolución 3D

def conv3D(imagen, filtro):

'''

Convolución 3D

: imagen param: input.shape is)

b = np.sum(imagen)

imagen_nueva[i, j] = np.sum([r, g, b])

imagen_nueva = imagen_nueva.clip(0, 255)

image_new = np.rint(image_ new).astype('uint8')

devuelve image_new