torchkbnufft
torchkbnufft copied to clipboard
mmuckley
A high-level, easy-to-deploy non-uniform Fast Fourier Transform in PyTorch.
Hi, as far as I can see, there is no support for real-valued input arrays? However, with `rfft` we could save a factor of 2 on computation for those cases...
Hi, I'm trying to perform on-the-fly data undersampling in my PyTorch dataset. To do this, I perform a Toeplitz NUFFT in the `__getitem__` function of my `Dataset` class. This works...
We should have some parameter that the user can pass that controls the number of threads used for a NUFFT. Presently we discover the number of threads on the system...
Hi, to be able to use `torch.func.*` (for example, `torch.func.grad`) with torchkbnufft, the` torch.autograd.Function`s in [torchkbnufft/_autograd/interp.py](https://github.com/mmuckley/torchkbnufft/blob/7039bc1c2d32cf86c8f25d39bcf1bf394368bdd4/torchkbnufft/_autograd/interp.py) would need an update to the newer calling signature. See https://pytorch.org/docs/stable/notes/extending.func.html. In particular, the...
