mlx.core.fft.rfftn

目录

mlx.core.fft.rfftn#

rfftn(a: array, s: Optional[Sequence[int]] = None, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#

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

输出的形状与输入相同，除了 axes 中的维度，在这种情况下，其大小取自 s。 axes 中的最后一个轴被视为实数轴，其大小将为 s[-1] // 2 + 1。

参数:

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

  • s (list(int), 可选) – 变换轴的大小。输入的相应轴将被截断或用零填充以匹配 s 中的大小。默认值是 a 沿 axes 的大小。

  • axes (list(int), 可选) – 执行 FFT 的轴。默认值为 None，在这种情况下，FFT 会作用于最后 len(s) 个轴，如果 s 也是 None，则作用于所有轴。

返回:

输入沿给定轴的实数 DFT。输出

返回类型:

数组

目录