enh: one less iterated operation (transpose)

nipy · Nov 16, 2023 · 8b21efa · 8b21efa
1 parent 3ec84d2
commit 8b21efa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nitransforms/linear.py b/nitransforms/linear.py
@@ -458,7 +458,7 @@ def apply(
         output_dtype = output_dtype or input_dtype
 
         # Prepare physical coordinates of input (grid, points)
-        xcoords = _ref.ndcoords.astype("f4")
+        xcoords = _ref.ndcoords.astype("f4").T
 
         # Invert target's (moving) affine once
         ras2vox = ~Affine(spatialimage.affine)
@@ -483,7 +483,7 @@ def apply(
 
         for t, xfm_t in enumerate(self):
             # Map the input coordinates on to timepoint t of the target (moving)
-            ycoords = xfm_t.map(xcoords.T)[..., : _ref.ndim]
+            ycoords = xfm_t.map(xcoords)[..., : _ref.ndim]
 
             # Calculate corresponding voxel coordinates
             yvoxels = ras2vox.map(ycoords)[..., : _ref.ndim]