mlx.core.fft.rfft2

rfft2(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = [-2, -1], stream: Optional[Union[Stream, Device]] = None) → array

Two dimensional real discrete Fourier Transform.

The output has the same shape as the input except along the dimensions in axes in which case it has sizes from s. The last axis in axes is treated as the real axis and will have size s[-1] // 2 + 1.

Parameters:

• a (array) – The input array. If the array is complex it will be silently cast to a real type.

• s (list(int), optional) – Sizes of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match the sizes in s. The default value is the sizes of a along axes.

• axes (list(int), optional) – Axes along which to perform the FFT. The default is [-2, -1].

Returns:

The real DFT of the input along the given axes. The output data type will be complex.

Return type:

array