mlx.core.fft.rfft

rfft(a: array, n: Optional[int] = None, axis: int = -1, stream: Optional[Union[Stream, Device]] = None) → array

对实数输入进行一维离散傅里叶变换。

输出的形状与输入相同，除了沿 axis 轴以外，在这种情况下其大小为 n // 2 + 1。

参数：

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

• n (int, 可选的) – 变换轴的大小。输入中对应的轴将被截断或用零填充以匹配 n。默认值为 a.shape[axis]。

• axis (int, 可选的) – 执行 FFT 的轴。默认值为 -1。

返回：

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

返回类型：

array