jax.scipy.fft.dctn¶

jax.scipy.fft.dctn(x, type=2, s=None, axes=None, norm=None)[source]¶

Return multidimensional Discrete Cosine Transform along the specified axes.

LAX-backend implementation of dctn().

Original docstring below.

Parameters

x (array_like) – The input array.
type ({1, 2, 3, 4}, optional) – Type of the DCT (see Notes). Default type is 2.
axes (int or array_like of ints or None, optional) – Axes along which the DCT is computed. The default is over all axes.
norm ({None, 'ortho'}, optional) – Normalization mode (see Notes). Default is None.

Returns

y – The transformed input array.

Return type

ndarray of real