jax.numpy.fft.irfft#
- jax.numpy.fft.irfft(a, n=None, axis=-1, norm=None)[source]#
计算实值一维逆离散傅里叶变换。
JAX 实现
numpy.fft.irfft()
.- 参数:
- 返回值:
一个包含
a
的一维逆离散傅里叶变换的实值数组,沿axis
轴的维度为n
。- 返回类型:
另请参阅
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)