minor bug fix

kelvin34501 · Nov 9, 2022 · 8f9cfdc · 8f9cfdc
1 parent 4be5fa0
commit 8f9cfdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmin/function.py b/torchmin/function.py
@@ -35,7 +35,7 @@ def __init__(self,
         self.gx = gx
         self.symmetric = symmetric
         # tensor-like properties
-        self.shape = (x.numel(), x.numel())
+        self.shape = (f.numel(), x.numel())
         self.dtype = x.dtype
         self.device = x.device