mlx.core.convolve

convolve(a: array, v: array, /, mode: str = 'full', *, stream: None | Stream | Device = None) → array

一维数组的离散卷积。

如果 v 比 a 长，则它们会被交换。卷积核按照信号处理约定进行翻转。

参数:

• a (array) – 一维输入数组。

• v (array) – 一维输入数组。

• mode (str, 可选) – {‘full’, ‘valid’, ‘same’}

返回:

卷积后的数组。

返回类型:

array