mmuckley / torchkbnufft

A high-level, easy-to-deploy non-uniform Fast Fourier Transform in PyTorch.
https://torchkbnufft.readthedocs.io/
MIT License
204 stars 44 forks source link

Add PyTorch 1.7 FFT compatibility #11

Closed mmuckley closed 3 years ago

mmuckley commented 3 years ago

This PR adds compatibility for the new PyTorch 1.7 complex number FFT. It handles it just by converting the real tensor to complex and back.