mlx.core.fft.irfftn

目录

mlx.core.fft.irfftn#

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

mlx.core.fft.rfftn() 的逆变换。

注意输入通常是复数。axes 中指定的输入维度会被填充或截断以匹配 s 中的大小。axes 中的最后一个轴被视为实轴,其大小将为 s[-1] // 2 + 1。

参数:
  • a (array) – 输入数组。

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

  • axes (list(int), optional) – 执行 FFT 的轴。默认值为 None,此时 FFT 将作用于最后的 len(s) 个轴,如果 s 也为 None,则作用于所有轴。

返回值:

包含 mlx.core.fft.rfftn() 逆变换的实数数组。

返回值类型:

array