add jacobian_var

pasqal-io · Apr 14, 2024 · ca975bf · ca975bf
1 parent 22db86c
commit ca975bf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/qadence_libs/qinfo_tools/utils.py b/qadence_libs/qinfo_tools/utils.py
@@ -28,10 +28,11 @@ def hessian(output: Tensor, inputs: list[Tensor]) -> Tensor:
     n_params = len(inputs)
     hess = torch.empty((n_params, n_params))
     for i in range(n_params):
+        jacobian_var = jacobian[i]
         ovrlp_grad2 = grad(
-            jacobian[i],
+            jacobian_var,
             inputs,
-            torch.ones_like(jacobian[i]),
+            torch.ones_like(jacobian_var),
             create_graph=True,
             allow_unused=True,
         )