Patch forward autograd - Githubissues

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

Patch forward autograd #23

Closed mmuckley closed 3 years ago

mmuckley commented 3 years ago

Fixes Issue #22. An incredibly weird bug that broke autograd that only occurred when calling forward multiple times, and it was caused by using unsqueeze operations.