JaxBackend.build_fft#

JaxBackend.build_fft(fast_shape, fast_ft_shape, inverse_fast_shape=None, **kwargs)[source]#

Build forward and inverse real fourier transform functions. The returned callables have two parameters arr and out which correspond to the input and output of the Fourier transform. The methods return the output of the respective function call, regardless of out being provided or not, analogous to most numpy 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).
real_dtypedtype: Numpy dtype of the inverse fourier transform.
complex_dtypedtype: Numpy dtype of the fourier transform.
inverse_fast_shapetuple, optional: Output shape of the inverse Fourier transform. By default fast_shape.
fftargsdict, optional: Dictionary passed to pyFFTW builders.
temp_realNDArray, optional: Temporary real numpy array, by default None.
temp_fftNDArray, optional: Temporary fft numpy array, by default None.

Returns:

tuple: Tuple of callables for forward and inverse real Fourier transform.