make DensityMatrixSimulator more numerically stable (#6850)

Co-authored-by: Pavol Juhas <[email protected]>
quantumlib · Dec 17, 2024 · 324a086 · 324a086
1 parent ea605d4
commit 324a086
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cirq-core/cirq/sim/simulation_utils.py b/cirq-core/cirq/sim/simulation_utils.py
@@ -56,4 +56,6 @@ def state_probabilities_by_indices(
         probs = np.sum(probs, axis=-1)
 
     # To deal with rounding issues, ensure that the probabilities sum to 1.
-    return probs / np.sum(probs)
+    probs = np.clip(probs, 0, None)
+    probs /= probs.sum()
+    return probs