Update _tensor.py

adtzlr · Jan 12, 2025 · 4a262e2 · 4a262e2
1 parent 743c1b4
commit 4a262e2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/tensortrax/_tensor.py b/src/tensortrax/_tensor.py
@@ -255,10 +255,11 @@ def init(self, gradient=False, hessian=False, sym=False, δx=None, Δx=None):
                     δx = np.zeros(self.size**2).reshape(shape)
                 else:
                     δx = np.eye(self.size).reshape(shape)
-                
+
             else:
                 δx = δx.reshape(*self.shape, *self.trax)
-                Δx = δx.copy()
+
+            Δx = δx.copy()
 
         elif hessian:
             # add additional trailing axes for dual values