jax.numpy.fft.ihfft

内容

jax.numpy.fft.ihfft#

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

计算一个频谱具有 Hermitian 对称性的数组的 1-D 逆 FFT。

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

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

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

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

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

返回值:

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

返回类型:

数组

另请参见

示例

>>> 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=4axis=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)