Hi everyone, I just wrote a library doing exactly this, you might want to check it out:

  
      
      GitHub
  
  
    

zh217/torch-dct 217

DCT (discrete cosine transform) functions for pytorch - zh217/torch-dct


  
  
    
    
  
  


Would love to hear your feedback!