Fix a bug in dense_to_mps

PennyLaneAI · Oct 10, 2024 · 74b813e · 74b813e
1 parent acd1f70
commit 74b813e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demonstrations/tutorial_mps.py b/demonstrations/tutorial_mps.py
@@ -330,7 +330,7 @@ def dense_to_mps(psi, bond_dim):
     Ms.append(U)
     Ss.append(Ss)
     bondL = Vd.shape[0]
-    psi = Vd
+    psi = np.tensordot(np.diag(S), Vd, 1)
 
     for _ in range(n-2):
         psi = np.reshape(psi, (2*bondL, -1)) # reshape psi[2 * bondL, (2x2x2...)]