dct_2d

deepinv.physics.functional.dct_2d(x, norm=None)

2-dimentional Discrete Cosine Transform, Type II (a.k.a. the DCT)

For the meaning of the parameter norm, see: https://docs.scipy.org/doc/scipy-0.14.0/reference/generated/scipy.fftpack.dct.html

Parts of this code are adapted from the torch-dct repository by zh217: zh217/torch-dct

Parameters:

• x (torch.Tensor) – the input signal

• norm (None, str) – the normalization, None or 'ortho'

Returns:

(torch.Tensor) the DCT-II of the signal over the last 2 dimensions