feat: PSR working but x is padded

qiboteam · Oct 28, 2024 · 687e843 · 687e843
1 parent c1534e1
commit 687e843
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 102 deletions.
diff --git a/exercise.py b/exercise.py
diff --git a/src/qiboml/models/pytorch.py b/src/qiboml/models/pytorch.py
@@ -24,7 +24,7 @@ def __post_init__(
         super().__init__()
 
         params = [p for param in self.circuit.get_parameters() for p in param]
-        params = torch.as_tensor(self.backend.to_numpy(params)).ravel()
+        params = torch.as_tensor(self.backend.to_numpy(x=params)).ravel()
         params.requires_grad = True
         self.circuit_parameters = torch.nn.Parameter(params)
 

diff --git a/src/qiboml/operations/differentiation.py b/src/qiboml/operations/differentiation.py
@@ -25,7 +25,7 @@ def evaluate(self, x: ndarray, encoding, training, decoding, backend, *parameter
                 "Parameter Shift Rule only supports expectation value decoding.",
             )
         x = encoding(x) + training
-        gradients = []
+        gradients = [np.array([[0.0]])]
         for i in range(len(parameters)):
             gradients.append(
                 self.one_parameter_shift(