mlx.core.convolve

目录

mlx.core.convolve#

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

The discrete convolution of 1D arrays.

If v is longer than a, then they are swapped. The conv filter is flipped following signal processing convention.

参数:
  • a (array) -- 1D Input array.

  • v (array) -- 1D Input array.

  • mode (str, optional) -- {'full', 'valid', 'same'}

返回:

The convolved array.

返回类型:

array