Fix a bug in dense_to_mps (#1237)

Bug reported by @hsim13372 , thanks a lot! Quick fix. This was missed because the function is just used to show the dimensionality.
PennyLaneAI · Oct 11, 2024 · 468266c · 468266c
1 parent acd1f70
commit 468266c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/demonstrations/tutorial_mps.metadata.json b/demonstrations/tutorial_mps.metadata.json
@@ -6,7 +6,7 @@
         }
     ],
     "dateOfPublication": "2024-09-29T00:00:00+00:00",
-    "dateOfLastModification": "2024-10-07T00:00:00+00:00",
+    "dateOfLastModification": "2024-10-11T00:00:00+00:00",
     "categories": [
         "Quantum Computing",
         "Algorithms",

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...)]