Merge pull request #769 from qiboteam/depolarizing_channel

Depolarizing channel on gpu
qiboteam · Jan 25, 2023 · 980e841 · 980e841
2 parents 4ed2fe6 + cfba10c
commit 980e841
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/qibo/gates/channels.py b/src/qibo/gates/channels.py
@@ -303,9 +303,9 @@ def __init__(self, q, lam=0):
 
     def apply_density_matrix(self, backend, state, nqubits):
         lam = self.init_kwargs["lam"]
-        return (1 - lam) * backend.cast(state) + lam / 2**nqubits * I(
-            *range(nqubits)
-        ).asmatrix(backend)
+        return (1 - lam) * backend.cast(state) + lam / 2**nqubits * backend.cast(
+            I(*range(nqubits)).asmatrix(backend)
+        )
 
     def apply(self, backend, state, nqubits):
         num_qubits = len(self.target_qubits)