jax.numpy.fft.irfft

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

計算實數值一維反離散傅立葉轉換。

JAX 版本的 numpy.fft.irfft() 實作。

參數：

• a (ArrayLike) – 輸入陣列。

• n (int | None | None) – 整數。指定沿 axis 的結果維度。如果未指定，則 n = 2*(m-1)，其中 m 是 a 沿 axis 的維度。

• axis (int) – 整數，預設值=-1。指定計算轉換的軸。如果未指定，則沿軸 -1 計算轉換。

• norm (str | None | None) – 字串。正規化模式。支援 “backward”、“ortho” 和 “forward”。

回傳值：

一個實數值陣列，包含 a 的一維反離散傅立葉轉換，沿 axis 的維度為 n。

回傳型別：

Array

參見

• jax.numpy.fft.ifft()：計算一維反離散傅立葉轉換。

• jax.numpy.fft.irfft()：計算實數輸入的一維反離散傅立葉轉換。

• jax.numpy.fft.rfftn()：計算實數輸入的多維離散傅立葉轉換。

• jax.numpy.fft.irfftn()：計算實數輸入的多維反離散傅立葉轉換。

範例

jnp.fft.rfft 預設沿著 axis -1 計算轉換。

>>> x = jnp.array([[1, 3, 5],
...                [2, 4, 6]])
>>> jnp.fft.irfft(x)
Array([[ 3., -1.,  0., -1.],
       [ 4., -1.,  0., -1.]], dtype=float32)

當 n=3 時，沿軸 -1 的轉換維度將為 3，而沿其他軸的維度將與輸入相同。

>>> with jnp.printoptions(precision=2, suppress=True):
...   jnp.fft.irfft(x, n=3)
Array([[ 2.33, -0.67, -0.67],
       [ 3.33, -0.67, -0.67]], dtype=float32)

當 n=4 且 axis=0 時，沿 axis 0 的轉換維度將為 4，而沿其他軸的維度將與輸入相同。

>>> with jnp.printoptions(precision=2, suppress=True):
...   jnp.fft.irfft(x, n=4, axis=0)
Array([[ 1.25,  2.75,  4.25],
       [ 0.25,  0.75,  1.25],
       [-0.75, -1.25, -1.75],
       [ 0.25,  0.75,  1.25]], dtype=float32)