DCT (discrete cosine transform) functions for pytorch - View it on GitHub