Skip to content

Commit ab730c6

Browse files
committed
Fix normalization of bottom_matrix in projector code
1 parent 6f31092 commit ab730c6

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

peps_ad/ctmrg/projectors.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -274,15 +274,15 @@ def _left_projectors_workhorse(
274274
_,
275275
) = _quarter_tensors_to_matrix(top_left, top_right, bottom_left, bottom_right)
276276
top_matrix /= jnp.linalg.norm(top_matrix)
277-
bottom_matrix /= jnp.linalg.norm(top_matrix)
277+
bottom_matrix /= jnp.linalg.norm(bottom_matrix)
278278
elif projector_method is Projector_Method.FISHMAN:
279279
top_U, top_S, _, _, bottom_S, bottom_Vh = _fishman_horizontal_cut(
280280
top_left, top_right, bottom_left, bottom_right, truncation_eps
281281
)
282282
top_matrix = top_U * jnp.sqrt(top_S)[jnp.newaxis, :]
283283
bottom_matrix = jnp.sqrt(bottom_S)[:, jnp.newaxis] * bottom_Vh
284284
top_matrix /= jnp.linalg.norm(top_matrix)
285-
bottom_matrix /= jnp.linalg.norm(top_matrix)
285+
bottom_matrix /= jnp.linalg.norm(bottom_matrix)
286286
else:
287287
raise ValueError("Invalid projector method!")
288288

@@ -381,15 +381,15 @@ def _right_projectors_workhorse(
381381
bottom_matrix,
382382
) = _quarter_tensors_to_matrix(top_left, top_right, bottom_left, bottom_right)
383383
top_matrix /= jnp.linalg.norm(top_matrix)
384-
bottom_matrix /= jnp.linalg.norm(top_matrix)
384+
bottom_matrix /= jnp.linalg.norm(bottom_matrix)
385385
elif projector_method is Projector_Method.FISHMAN:
386386
_, top_S, top_Vh, bottom_U, bottom_S, _ = _fishman_horizontal_cut(
387387
top_left, top_right, bottom_left, bottom_right, truncation_eps
388388
)
389389
top_matrix = jnp.sqrt(top_S)[:, jnp.newaxis] * top_Vh
390390
bottom_matrix = bottom_U * jnp.sqrt(bottom_S)[jnp.newaxis, :]
391391
top_matrix /= jnp.linalg.norm(top_matrix)
392-
bottom_matrix /= jnp.linalg.norm(top_matrix)
392+
bottom_matrix /= jnp.linalg.norm(bottom_matrix)
393393
else:
394394
raise ValueError("Invalid projector method!")
395395

0 commit comments

Comments
 (0)