jax.scipy.fft.dct

jax.scipy.fft.dct(x, type=2, n=None, axis=-1, norm=None)

计算输入的离散余弦变换

scipy.fft.dct() 的 JAX 实现。

参数:

• x (Array) – 数组

• type (int) – 整数，默认值 = 2。目前仅支持 type 2。

• n (int | None | None) – 整数，默认值 = x.shape[axis]。变换的长度。如果大于 x.shape[axis]，则输入将进行零填充；如果小于，则输入将被截断。

• axis (int) – 整数，默认值=-1。执行 dct 的轴。

• norm (str | None | None) – 字符串。归一化模式： [None, "backward", "ortho"]之一。默认值为 None，等效于 "backward"。

返回:

包含 x 的离散余弦变换的数组

返回类型:

Array

另请参阅

• jax.scipy.fft.dctn(): 多维 DCT

• jax.scipy.fft.idct(): 反向 DCT

• jax.scipy.fft.idctn(): 多维反向 DCT

示例

>>> x = jax.random.normal(jax.random.key(0), (3, 3))
>>> with jnp.printoptions(precision=2, suppress=True):
...   print(jax.scipy.fft.dct(x))
[[ 6.43  3.56 -2.86]
 [-1.75  1.55 -1.4 ]
 [ 1.33 -2.01 -0.82]]

当 n 小于 x.shape[axis] 时

>>> with jnp.printoptions(precision=2, suppress=True):
...   print(jax.scipy.fft.dct(x, n=2))
[[ 7.3  -0.57]
 [ 0.19 -0.36]
 [-0.   -1.4 ]]

当 n 小于 x.shape[axis] 且 axis=0 时

>>> with jnp.printoptions(precision=2, suppress=True):
...   print(jax.scipy.fft.dct(x, n=2, axis=0))
[[ 3.09  4.4  -2.81]
 [ 2.41  2.62  0.76]]

当 n 大于 x.shape[axis] 且 axis=1 时

>>> with jnp.printoptions(precision=2, suppress=True):
...   print(jax.scipy.fft.dct(x, n=4, axis=1))
[[ 6.43  4.88  0.04 -3.3 ]
 [-1.75  0.73  1.01 -2.18]
 [ 1.33 -1.05 -2.34 -0.07]]