Patch forward autograd (#23)

* Patch forward autograd * Increment version
mmuckley · Feb 4, 2021 · b5f6581 · b5f6581
1 parent d568c43
commit b5f6581
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/torchkbnufft/__init__.py b/torchkbnufft/__init__.py
@@ -1,6 +1,6 @@
 """Package info"""
 
-__version__ = "1.0.0"
+__version__ = "1.0.1"
 __author__ = "Matthew Muckley"
 __author_email__ = "matt.muckley@gmail.com"
 __license__ = "MIT"

diff --git a/torchkbnufft/_nufft/fft.py b/torchkbnufft/_nufft/fft.py
@@ -68,8 +68,6 @@ def fft_and_scale(
         pad_sizes.append(0)
         pad_sizes.append(int(gd - im))
 
-    scaling_coef = scaling_coef.unsqueeze(0).unsqueeze(0)
-
     # multiply by scaling_coef, pad, then fft
     return fft_fn(
         F.pad(image * scaling_coef, pad_sizes),
@@ -111,8 +109,6 @@ def ifft_and_scale(
     # calculate crops
     dims = torch.arange(len(im_size), device=image.device) + 2
 
-    scaling_coef = scaling_coef.unsqueeze(0).unsqueeze(0)
-
     # ifft, crop, then multiply by scaling_coef conjugate
     return (
         crop_dims(