Fixed a tiny bug.

WeiguoMa · WeiguoMa · commit e6f520899967 · 2025-08-14T18:59:32.000+08:00
diff --git a/tensorcircuit/mpscircuit.py b/tensorcircuit/mpscircuit.py
@@ -1077,7 +1077,7 @@ def measure(
                 backend.convert_to_tensor(np.array([1.0], dtype=dtypestr)),
             )
 
-            p = p * backend.sum(ps * m)
+            p = p * backend.sum(ps * backend.cast(m, dtype=rdtypestr))
             mps._mps.tensors[site] = backend.einsum("iaj,a->ij", tensor, m)[:, None, :]
         sample = backend.stack(sample)
         sample = backend.real(sample)

Original file line number	Diff line number	Diff line change
`@@ -1077,7 +1077,7 @@ def measure(`
`1077`	`1077`	`backend.convert_to_tensor(np.array([1.0], dtype=dtypestr)),`
`1078`	`1078`	`)`
`1079`	`1079`
`1080`		`- p = p * backend.sum(ps * m)`
	`1080`	`+ p = p * backend.sum(ps * backend.cast(m, dtype=rdtypestr))`
`1081`	`1081`	`mps._mps.tensors[site] = backend.einsum("iaj,a->ij", tensor, m)[:, None, :]`
`1082`	`1082`	`sample = backend.stack(sample)`
`1083`	`1083`	`sample = backend.real(sample)`