mlx.core.fft.irfft2

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

rfft2() 的逆变换。

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

参数:

• a (array) – 输入数组。

• s (list(int), 可选) – 转换轴的大小。输入中相应的轴会被截断或用零填充，以匹配 s 中的尺寸，除了最后一个轴的大小为 s[-1] // 2 + 1。默认值是 a 沿 axes 的尺寸。

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

返回值:

包含 rfft2() 逆变换的实数数组。

返回类型:

array