From ca975bff1991e7be7011469a1935b19a56e87ff0 Mon Sep 17 00:00:00 2001 From: Ignacio Date: Sun, 14 Apr 2024 11:27:15 +0200 Subject: [PATCH] add jacobian_var --- qadence_libs/qinfo_tools/utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/qadence_libs/qinfo_tools/utils.py b/qadence_libs/qinfo_tools/utils.py index fb5ff62..431d8f0 100644 --- 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, )