jax.numpy.fft.ihfft

jax.numpy.fft.ihfft(a, n=None, axis=-1, norm=None)

计算具有厄米对称频谱的数组的一维逆FFT。

JAX 实现的 numpy.fft.ihfft()。

参数:

• a (ArrayLike) – 输入数组。

• n (int | None) – 可选，整型。指定输入沿 axis 的有效维度。如果未指定，则默认为输入沿 axis 的维度。

• axis (int) – 可选，整型，默认为 -1。指定计算变换的轴。如果未指定，则变换将沿轴 -1 计算。

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

返回:

一个数组，其中包含通过利用其固有的厄米对称性得到的 a 的一维离散傅里叶变换。如果 n 是偶数，则数组沿 axis 的维度为 (n/2)+1；如果 n 是奇数，则为 (n+1)/2。

返回类型:

数组

另请参阅

• jax.numpy.fft.hfft(): 计算具有厄米对称频谱的数组的一维FFT。

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

• jax.numpy.fft.rfft(): 计算实值输入的一维离散傅里叶变换。

示例

>>> x = jnp.array([[1, 3, 5, 7],
...                [2, 4, 6, 8]])
>>> jnp.fft.ihfft(x)
Array([[ 4.+0.j, -1.-1.j, -1.-0.j],
       [ 5.+0.j, -1.-1.j, -1.-0.j]], dtype=complex64)

当 n=4 且 axis=0 时，沿 axis 0 的变换维度将为 (4/2)+1 =3，而沿其他轴的维度将与输入相同。

>>> jnp.fft.ihfft(x, n=4, axis=0)
Array([[ 0.75+0.j ,  1.75+0.j ,  2.75+0.j ,  3.75+0.j ],
       [ 0.25+0.5j,  0.75+1.j ,  1.25+1.5j,  1.75+2.j ],
       [-0.25-0.j , -0.25-0.j , -0.25-0.j , -0.25-0.j ]], dtype=complex64)