jax.numpy.fft.irfftn

jax.numpy.fft.irfftn(a, s=None, axes=None, norm=None)

计算实值多维离散傅里叶逆变换。

JAX 实现 numpy.fft.irfftn()。

参数:

• a (ArrayLike) – 输入数组。

• s (形状 | None) – 可选整数序列。指定输出在每个指定轴上的大小。如果未指定，则沿 axes[-1] 轴的输出尺寸为 2*(m-1)，其中 m 是沿 axes[-1] 轴的输入尺寸，其他轴的尺寸将与输入相同。

• axes (序列[int] | None) – 可选整数序列，默认为 None。指定计算变换的轴。如果未指定，则沿最后 len(s) 个轴计算变换。如果 axes 和 s 都未指定，则沿所有轴计算变换。

• norm (str | None) – 字符串，默认为“backward”。归一化模式。支持“backward”、“ortho”和“forward”。

返回:

一个实值数组，包含 a 的多维离散傅里叶逆变换，在指定的 axes 上大小为 s，在其他轴上与输入相同。

返回类型:

Array

另请参阅

• jax.numpy.fft.rfftn(): 计算实值数组的多维离散傅里叶变换。

• jax.numpy.fft.irfft(): 计算实值一维离散傅里叶逆变换。

• jax.numpy.fft.irfft2(): 计算实值二维离散傅里叶逆变换。

示例

jnp.fft.irfftn 默认沿所有轴计算变换。

>>> x = jnp.array([[[1, 3, 5],
...                 [2, 4, 6]],
...                [[7, 9, 11],
...                 [8, 10, 12]]])
>>> jnp.fft.irfftn(x)
Array([[[ 6.5, -1. ,  0. , -1. ],
        [-0.5,  0. ,  0. ,  0. ]],

       [[-3. ,  0. ,  0. ,  0. ],
        [ 0. ,  0. ,  0. ,  0. ]]], dtype=float32)

当 s=[3, 4] 时，沿 axes (-2, -1) 的变换大小将为 (3, 4)，其他轴的大小将与输入相同。

>>> with jnp.printoptions(precision=2, suppress=True):
...   jnp.fft.irfftn(x, s=[3, 4])
Array([[[ 2.33, -0.67,  0.  , -0.67],
        [ 0.33, -0.74,  0.  ,  0.41],
        [ 0.33,  0.41,  0.  , -0.74]],

       [[ 6.33, -0.67,  0.  , -0.67],
        [ 1.33, -1.61,  0.  ,  1.28],
        [ 1.33,  1.28,  0.  , -1.61]]], dtype=float32)

当 s=[3] 且 axes=[0] 时，沿 axes 0 的变换大小将为 3，其他轴的尺寸将与输入相同。

>>> with jnp.printoptions(precision=2, suppress=True):
...   jnp.fft.irfftn(x, s=[3], axes=[0])
Array([[[ 5.,  7.,  9.],
        [ 6.,  8., 10.]],

       [[-2., -2., -2.],
        [-2., -2., -2.]],

       [[-2., -2., -2.],
        [-2., -2., -2.]]], dtype=float32)