Fixing grad

BQSKit · Sep 27, 2024 · 63a6795 · 63a6795
1 parent 6c08005
commit 63a6795
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bqskit/ir/gates/parameterized/diagonal.py b/bqskit/ir/gates/parameterized/diagonal.py
@@ -59,7 +59,7 @@ def get_grad(self, params: RealVector = []) -> npt.NDArray[np.complex128]:
         )
 
         for i, ind in enumerate(range(1, 2 ** self.num_qudits)):
-            grad[ind][i][i] = 1j * np.exp(1j * params[ind])
+            grad[i][ind][ind] = 1j * np.exp(1j * params[i])
 
         return grad