MLXBackend.build_fft#

MLXBackend.build_fft(fast_shape, fast_ft_shape, **kwargs)[source]#

Build fft builder functions.

Parameters:
fast_shapetuple

Tuple of integers corresponding to fast convolution shape (see compute_convolution_shapes).

fast_ft_shapetuple

Tuple of integers corresponding to the shape of the fourier transform array (see compute_convolution_shapes).

**kwargsdict, optional

Additional parameters that are not used for now.

Returns:
tuple

Tupple containing callable rfft and irfft object.