torch.rfft lacks of doc and it’s hard to understand how to use it.
Actually, I’d like to use this function to implement a fast discrete cosine transform (DCT).
Please let me know if you have DCT implementations (any differentiable in PyTorch) or concrete example for torch.rfft (especially, 2D case).
Thanks.