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