mlx.core.fft.rfft2

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

二维实数离散傅里叶变换。

输出的形状与输入相同，但在 axes 指定的维度上除外，这些维度的大小由 s 决定。 axes 中的最后一个轴被视为实数轴，其大小将为 s[-1] // 2 + 1。

参数:

• a (array) – 输入数组。如果数组是复数类型，它将被隐式转换为实数类型。

• s (list(int), optional) – 变换轴的大小。输入的相应轴将被截断或用零填充以匹配 s 中的大小。默认值是输入数组 a 在 axes 指定轴上的大小。

• axes (list(int), optional) – 执行 FFT 的轴。默认值为 [-2, -1]。

返回:

输入在给定轴上的实数 DFT。输出数据类型将是复数类型。

返回类型:

array