16. Real time simple update (SU)ΒΆ

%config InlineBackend.figure_formats = ['svg']
import autoray as ar
import matplotlib as mpl
import matplotlib.pyplot as plt
import tqdm  # for progress bar

import quimb as qu
import quimb.tensor as qtn

mpl.style.use(qu.NEUTRAL_STYLE)
Lx = 11
Ly = 11

edges = qtn.edges_2d_square(Lx, Ly, cyclic=False)
sites = sorted({s for e in edges for s in e})

# heisenberg interaction
h2 = -qu.pauli("X") & qu.pauli("X")
# quenched random onsite z-fields
h1 = -3.04438 * qu.pauli("Z")
# all zero-state
psi = qtn.TN_from_sites_product_state(
    site_map={site: [1.0, 0.0] for site in sites}
)

# make size 1 lattice bonds
for cooa, coob in edges:
    psi[cooa].new_bond(psi[coob])

psi
TensorNetworkGenVector(tensors=121, indices=341)
Tensor(shape=(1, 1, 2), inds=[_3a1604AAAAB, _3a1604AAAAA, k(0, 0)], tags={I(0, 0)}),backend=numpy, dtype=float64, data=array([[[1., 0.]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAD, _3a1604AAAAC, _3a1604AAAAA, k(0, 1)], tags={I(0, 1)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAF, _3a1604AAAAE, _3a1604AAAAC, k(0, 2)], tags={I(0, 2)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAH, _3a1604AAAAG, _3a1604AAAAE, k(0, 3)], tags={I(0, 3)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAJ, _3a1604AAAAI, _3a1604AAAAG, k(0, 4)], tags={I(0, 4)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAL, _3a1604AAAAK, _3a1604AAAAI, k(0, 5)], tags={I(0, 5)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAN, _3a1604AAAAM, _3a1604AAAAK, k(0, 6)], tags={I(0, 6)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAP, _3a1604AAAAO, _3a1604AAAAM, k(0, 7)], tags={I(0, 7)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAR, _3a1604AAAAQ, _3a1604AAAAO, k(0, 8)], tags={I(0, 8)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAT, _3a1604AAAAS, _3a1604AAAAQ, k(0, 9)], tags={I(0, 9)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 2), inds=[_3a1604AAAAU, _3a1604AAAAS, k(0, 10)], tags={I(0, 10)}),backend=numpy, dtype=float64, data=array([[[1., 0.]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAW, _3a1604AAAAV, _3a1604AAAAB, k(1, 0)], tags={I(1, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAY, _3a1604AAAAX, _3a1604AAAAV, _3a1604AAAAD, k(1, 1)], tags={I(1, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAa, _3a1604AAAAZ, _3a1604AAAAX, _3a1604AAAAF, k(1, 2)], tags={I(1, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAc, _3a1604AAAAb, _3a1604AAAAZ, _3a1604AAAAH, k(1, 3)], tags={I(1, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAe, _3a1604AAAAd, _3a1604AAAAb, _3a1604AAAAJ, k(1, 4)], tags={I(1, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAg, _3a1604AAAAf, _3a1604AAAAd, _3a1604AAAAL, k(1, 5)], tags={I(1, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAi, _3a1604AAAAh, _3a1604AAAAf, _3a1604AAAAN, k(1, 6)], tags={I(1, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAk, _3a1604AAAAj, _3a1604AAAAh, _3a1604AAAAP, k(1, 7)], tags={I(1, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAm, _3a1604AAAAl, _3a1604AAAAj, _3a1604AAAAR, k(1, 8)], tags={I(1, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAo, _3a1604AAAAn, _3a1604AAAAl, _3a1604AAAAT, k(1, 9)], tags={I(1, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAp, _3a1604AAAAn, _3a1604AAAAU, k(1, 10)], tags={I(1, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAr, _3a1604AAAAq, _3a1604AAAAW, k(2, 0)], tags={I(2, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAt, _3a1604AAAAs, _3a1604AAAAq, _3a1604AAAAY, k(2, 1)], tags={I(2, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAv, _3a1604AAAAu, _3a1604AAAAs, _3a1604AAAAa, k(2, 2)], tags={I(2, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAx, _3a1604AAAAw, _3a1604AAAAu, _3a1604AAAAc, k(2, 3)], tags={I(2, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAz, _3a1604AAAAy, _3a1604AAAAw, _3a1604AAAAe, k(2, 4)], tags={I(2, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABB, _3a1604AAABA, _3a1604AAAAy, _3a1604AAAAg, k(2, 5)], tags={I(2, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABD, _3a1604AAABC, _3a1604AAABA, _3a1604AAAAi, k(2, 6)], tags={I(2, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABF, _3a1604AAABE, _3a1604AAABC, _3a1604AAAAk, k(2, 7)], tags={I(2, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABH, _3a1604AAABG, _3a1604AAABE, _3a1604AAAAm, k(2, 8)], tags={I(2, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABJ, _3a1604AAABI, _3a1604AAABG, _3a1604AAAAo, k(2, 9)], tags={I(2, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAABK, _3a1604AAABI, _3a1604AAAAp, k(2, 10)], tags={I(2, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAABM, _3a1604AAABL, _3a1604AAAAr, k(3, 0)], tags={I(3, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABO, _3a1604AAABN, _3a1604AAABL, _3a1604AAAAt, k(3, 1)], tags={I(3, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABQ, _3a1604AAABP, _3a1604AAABN, _3a1604AAAAv, k(3, 2)], tags={I(3, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABS, _3a1604AAABR, _3a1604AAABP, _3a1604AAAAx, k(3, 3)], tags={I(3, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABU, _3a1604AAABT, _3a1604AAABR, _3a1604AAAAz, k(3, 4)], tags={I(3, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABW, _3a1604AAABV, _3a1604AAABT, _3a1604AAABB, k(3, 5)], tags={I(3, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABY, _3a1604AAABX, _3a1604AAABV, _3a1604AAABD, k(3, 6)], tags={I(3, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABa, _3a1604AAABZ, _3a1604AAABX, _3a1604AAABF, k(3, 7)], tags={I(3, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABc, _3a1604AAABb, _3a1604AAABZ, _3a1604AAABH, k(3, 8)], tags={I(3, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABe, _3a1604AAABd, _3a1604AAABb, _3a1604AAABJ, k(3, 9)], tags={I(3, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAABf, _3a1604AAABd, _3a1604AAABK, k(3, 10)], tags={I(3, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAABh, _3a1604AAABg, _3a1604AAABM, k(4, 0)], tags={I(4, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABj, _3a1604AAABi, _3a1604AAABg, _3a1604AAABO, k(4, 1)], tags={I(4, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABl, _3a1604AAABk, _3a1604AAABi, _3a1604AAABQ, k(4, 2)], tags={I(4, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABn, _3a1604AAABm, _3a1604AAABk, _3a1604AAABS, k(4, 3)], tags={I(4, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABp, _3a1604AAABo, _3a1604AAABm, _3a1604AAABU, k(4, 4)], tags={I(4, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABr, _3a1604AAABq, _3a1604AAABo, _3a1604AAABW, k(4, 5)], tags={I(4, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABt, _3a1604AAABs, _3a1604AAABq, _3a1604AAABY, k(4, 6)], tags={I(4, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABv, _3a1604AAABu, _3a1604AAABs, _3a1604AAABa, k(4, 7)], tags={I(4, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABx, _3a1604AAABw, _3a1604AAABu, _3a1604AAABc, k(4, 8)], tags={I(4, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABz, _3a1604AAABy, _3a1604AAABw, _3a1604AAABe, k(4, 9)], tags={I(4, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACA, _3a1604AAABy, _3a1604AAABf, k(4, 10)], tags={I(4, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACC, _3a1604AAACB, _3a1604AAABh, k(5, 0)], tags={I(5, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACE, _3a1604AAACD, _3a1604AAACB, _3a1604AAABj, k(5, 1)], tags={I(5, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACG, _3a1604AAACF, _3a1604AAACD, _3a1604AAABl, k(5, 2)], tags={I(5, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACI, _3a1604AAACH, _3a1604AAACF, _3a1604AAABn, k(5, 3)], tags={I(5, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACK, _3a1604AAACJ, _3a1604AAACH, _3a1604AAABp, k(5, 4)], tags={I(5, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACM, _3a1604AAACL, _3a1604AAACJ, _3a1604AAABr, k(5, 5)], tags={I(5, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACO, _3a1604AAACN, _3a1604AAACL, _3a1604AAABt, k(5, 6)], tags={I(5, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACQ, _3a1604AAACP, _3a1604AAACN, _3a1604AAABv, k(5, 7)], tags={I(5, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACS, _3a1604AAACR, _3a1604AAACP, _3a1604AAABx, k(5, 8)], tags={I(5, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACU, _3a1604AAACT, _3a1604AAACR, _3a1604AAABz, k(5, 9)], tags={I(5, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACV, _3a1604AAACT, _3a1604AAACA, k(5, 10)], tags={I(5, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACX, _3a1604AAACW, _3a1604AAACC, k(6, 0)], tags={I(6, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACZ, _3a1604AAACY, _3a1604AAACW, _3a1604AAACE, k(6, 1)], tags={I(6, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACb, _3a1604AAACa, _3a1604AAACY, _3a1604AAACG, k(6, 2)], tags={I(6, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACd, _3a1604AAACc, _3a1604AAACa, _3a1604AAACI, k(6, 3)], tags={I(6, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACf, _3a1604AAACe, _3a1604AAACc, _3a1604AAACK, k(6, 4)], tags={I(6, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACh, _3a1604AAACg, _3a1604AAACe, _3a1604AAACM, k(6, 5)], tags={I(6, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACj, _3a1604AAACi, _3a1604AAACg, _3a1604AAACO, k(6, 6)], tags={I(6, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACl, _3a1604AAACk, _3a1604AAACi, _3a1604AAACQ, k(6, 7)], tags={I(6, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACn, _3a1604AAACm, _3a1604AAACk, _3a1604AAACS, k(6, 8)], tags={I(6, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACp, _3a1604AAACo, _3a1604AAACm, _3a1604AAACU, k(6, 9)], tags={I(6, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACq, _3a1604AAACo, _3a1604AAACV, k(6, 10)], tags={I(6, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACs, _3a1604AAACr, _3a1604AAACX, k(7, 0)], tags={I(7, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACu, _3a1604AAACt, _3a1604AAACr, _3a1604AAACZ, k(7, 1)], tags={I(7, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACw, _3a1604AAACv, _3a1604AAACt, _3a1604AAACb, k(7, 2)], tags={I(7, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACy, _3a1604AAACx, _3a1604AAACv, _3a1604AAACd, k(7, 3)], tags={I(7, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADA, _3a1604AAACz, _3a1604AAACx, _3a1604AAACf, k(7, 4)], tags={I(7, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADC, _3a1604AAADB, _3a1604AAACz, _3a1604AAACh, k(7, 5)], tags={I(7, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADE, _3a1604AAADD, _3a1604AAADB, _3a1604AAACj, k(7, 6)], tags={I(7, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADG, _3a1604AAADF, _3a1604AAADD, _3a1604AAACl, k(7, 7)], tags={I(7, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADI, _3a1604AAADH, _3a1604AAADF, _3a1604AAACn, k(7, 8)], tags={I(7, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADK, _3a1604AAADJ, _3a1604AAADH, _3a1604AAACp, k(7, 9)], tags={I(7, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAADL, _3a1604AAADJ, _3a1604AAACq, k(7, 10)], tags={I(7, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAADN, _3a1604AAADM, _3a1604AAACs, k(8, 0)], tags={I(8, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADP, _3a1604AAADO, _3a1604AAADM, _3a1604AAACu, k(8, 1)], tags={I(8, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADR, _3a1604AAADQ, _3a1604AAADO, _3a1604AAACw, k(8, 2)], tags={I(8, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADT, _3a1604AAADS, _3a1604AAADQ, _3a1604AAACy, k(8, 3)], tags={I(8, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADV, _3a1604AAADU, _3a1604AAADS, _3a1604AAADA, k(8, 4)], tags={I(8, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADX, _3a1604AAADW, _3a1604AAADU, _3a1604AAADC, k(8, 5)], tags={I(8, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADZ, _3a1604AAADY, _3a1604AAADW, _3a1604AAADE, k(8, 6)], tags={I(8, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADb, _3a1604AAADa, _3a1604AAADY, _3a1604AAADG, k(8, 7)], tags={I(8, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADd, _3a1604AAADc, _3a1604AAADa, _3a1604AAADI, k(8, 8)], tags={I(8, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADf, _3a1604AAADe, _3a1604AAADc, _3a1604AAADK, k(8, 9)], tags={I(8, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAADg, _3a1604AAADe, _3a1604AAADL, k(8, 10)], tags={I(8, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAADi, _3a1604AAADh, _3a1604AAADN, k(9, 0)], tags={I(9, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])

...

# create simple update gauges
gauges = {}
psi.gauge_all_simple_(gauges=gauges)
TensorNetworkGenVector(tensors=121, indices=341)
Tensor(shape=(1, 1, 2), inds=[_3a1604AAAAB, _3a1604AAAAA, k(0, 0)], tags={I(0, 0)}),backend=numpy, dtype=float64, data=array([[[1., 0.]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAD, _3a1604AAAAC, _3a1604AAAAA, k(0, 1)], tags={I(0, 1)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAF, _3a1604AAAAE, _3a1604AAAAC, k(0, 2)], tags={I(0, 2)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAH, _3a1604AAAAG, _3a1604AAAAE, k(0, 3)], tags={I(0, 3)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAJ, _3a1604AAAAI, _3a1604AAAAG, k(0, 4)], tags={I(0, 4)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAL, _3a1604AAAAK, _3a1604AAAAI, k(0, 5)], tags={I(0, 5)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAN, _3a1604AAAAM, _3a1604AAAAK, k(0, 6)], tags={I(0, 6)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAP, _3a1604AAAAO, _3a1604AAAAM, k(0, 7)], tags={I(0, 7)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAR, _3a1604AAAAQ, _3a1604AAAAO, k(0, 8)], tags={I(0, 8)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAT, _3a1604AAAAS, _3a1604AAAAQ, k(0, 9)], tags={I(0, 9)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 2), inds=[_3a1604AAAAU, _3a1604AAAAS, k(0, 10)], tags={I(0, 10)}),backend=numpy, dtype=float64, data=array([[[1., 0.]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAW, _3a1604AAAAV, _3a1604AAAAB, k(1, 0)], tags={I(1, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAY, _3a1604AAAAX, _3a1604AAAAV, _3a1604AAAAD, k(1, 1)], tags={I(1, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAa, _3a1604AAAAZ, _3a1604AAAAX, _3a1604AAAAF, k(1, 2)], tags={I(1, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAc, _3a1604AAAAb, _3a1604AAAAZ, _3a1604AAAAH, k(1, 3)], tags={I(1, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAe, _3a1604AAAAd, _3a1604AAAAb, _3a1604AAAAJ, k(1, 4)], tags={I(1, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAg, _3a1604AAAAf, _3a1604AAAAd, _3a1604AAAAL, k(1, 5)], tags={I(1, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAi, _3a1604AAAAh, _3a1604AAAAf, _3a1604AAAAN, k(1, 6)], tags={I(1, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAk, _3a1604AAAAj, _3a1604AAAAh, _3a1604AAAAP, k(1, 7)], tags={I(1, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAm, _3a1604AAAAl, _3a1604AAAAj, _3a1604AAAAR, k(1, 8)], tags={I(1, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAo, _3a1604AAAAn, _3a1604AAAAl, _3a1604AAAAT, k(1, 9)], tags={I(1, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAp, _3a1604AAAAn, _3a1604AAAAU, k(1, 10)], tags={I(1, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAAAr, _3a1604AAAAq, _3a1604AAAAW, k(2, 0)], tags={I(2, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAt, _3a1604AAAAs, _3a1604AAAAq, _3a1604AAAAY, k(2, 1)], tags={I(2, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAv, _3a1604AAAAu, _3a1604AAAAs, _3a1604AAAAa, k(2, 2)], tags={I(2, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAx, _3a1604AAAAw, _3a1604AAAAu, _3a1604AAAAc, k(2, 3)], tags={I(2, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAAAz, _3a1604AAAAy, _3a1604AAAAw, _3a1604AAAAe, k(2, 4)], tags={I(2, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABB, _3a1604AAABA, _3a1604AAAAy, _3a1604AAAAg, k(2, 5)], tags={I(2, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABD, _3a1604AAABC, _3a1604AAABA, _3a1604AAAAi, k(2, 6)], tags={I(2, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABF, _3a1604AAABE, _3a1604AAABC, _3a1604AAAAk, k(2, 7)], tags={I(2, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABH, _3a1604AAABG, _3a1604AAABE, _3a1604AAAAm, k(2, 8)], tags={I(2, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABJ, _3a1604AAABI, _3a1604AAABG, _3a1604AAAAo, k(2, 9)], tags={I(2, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAABK, _3a1604AAABI, _3a1604AAAAp, k(2, 10)], tags={I(2, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAABM, _3a1604AAABL, _3a1604AAAAr, k(3, 0)], tags={I(3, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABO, _3a1604AAABN, _3a1604AAABL, _3a1604AAAAt, k(3, 1)], tags={I(3, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABQ, _3a1604AAABP, _3a1604AAABN, _3a1604AAAAv, k(3, 2)], tags={I(3, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABS, _3a1604AAABR, _3a1604AAABP, _3a1604AAAAx, k(3, 3)], tags={I(3, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABU, _3a1604AAABT, _3a1604AAABR, _3a1604AAAAz, k(3, 4)], tags={I(3, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABW, _3a1604AAABV, _3a1604AAABT, _3a1604AAABB, k(3, 5)], tags={I(3, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABY, _3a1604AAABX, _3a1604AAABV, _3a1604AAABD, k(3, 6)], tags={I(3, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABa, _3a1604AAABZ, _3a1604AAABX, _3a1604AAABF, k(3, 7)], tags={I(3, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABc, _3a1604AAABb, _3a1604AAABZ, _3a1604AAABH, k(3, 8)], tags={I(3, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABe, _3a1604AAABd, _3a1604AAABb, _3a1604AAABJ, k(3, 9)], tags={I(3, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAABf, _3a1604AAABd, _3a1604AAABK, k(3, 10)], tags={I(3, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAABh, _3a1604AAABg, _3a1604AAABM, k(4, 0)], tags={I(4, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABj, _3a1604AAABi, _3a1604AAABg, _3a1604AAABO, k(4, 1)], tags={I(4, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABl, _3a1604AAABk, _3a1604AAABi, _3a1604AAABQ, k(4, 2)], tags={I(4, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABn, _3a1604AAABm, _3a1604AAABk, _3a1604AAABS, k(4, 3)], tags={I(4, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABp, _3a1604AAABo, _3a1604AAABm, _3a1604AAABU, k(4, 4)], tags={I(4, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABr, _3a1604AAABq, _3a1604AAABo, _3a1604AAABW, k(4, 5)], tags={I(4, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABt, _3a1604AAABs, _3a1604AAABq, _3a1604AAABY, k(4, 6)], tags={I(4, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABv, _3a1604AAABu, _3a1604AAABs, _3a1604AAABa, k(4, 7)], tags={I(4, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABx, _3a1604AAABw, _3a1604AAABu, _3a1604AAABc, k(4, 8)], tags={I(4, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAABz, _3a1604AAABy, _3a1604AAABw, _3a1604AAABe, k(4, 9)], tags={I(4, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACA, _3a1604AAABy, _3a1604AAABf, k(4, 10)], tags={I(4, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACC, _3a1604AAACB, _3a1604AAABh, k(5, 0)], tags={I(5, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACE, _3a1604AAACD, _3a1604AAACB, _3a1604AAABj, k(5, 1)], tags={I(5, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACG, _3a1604AAACF, _3a1604AAACD, _3a1604AAABl, k(5, 2)], tags={I(5, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACI, _3a1604AAACH, _3a1604AAACF, _3a1604AAABn, k(5, 3)], tags={I(5, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACK, _3a1604AAACJ, _3a1604AAACH, _3a1604AAABp, k(5, 4)], tags={I(5, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACM, _3a1604AAACL, _3a1604AAACJ, _3a1604AAABr, k(5, 5)], tags={I(5, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACO, _3a1604AAACN, _3a1604AAACL, _3a1604AAABt, k(5, 6)], tags={I(5, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACQ, _3a1604AAACP, _3a1604AAACN, _3a1604AAABv, k(5, 7)], tags={I(5, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACS, _3a1604AAACR, _3a1604AAACP, _3a1604AAABx, k(5, 8)], tags={I(5, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACU, _3a1604AAACT, _3a1604AAACR, _3a1604AAABz, k(5, 9)], tags={I(5, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACV, _3a1604AAACT, _3a1604AAACA, k(5, 10)], tags={I(5, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACX, _3a1604AAACW, _3a1604AAACC, k(6, 0)], tags={I(6, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACZ, _3a1604AAACY, _3a1604AAACW, _3a1604AAACE, k(6, 1)], tags={I(6, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACb, _3a1604AAACa, _3a1604AAACY, _3a1604AAACG, k(6, 2)], tags={I(6, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACd, _3a1604AAACc, _3a1604AAACa, _3a1604AAACI, k(6, 3)], tags={I(6, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACf, _3a1604AAACe, _3a1604AAACc, _3a1604AAACK, k(6, 4)], tags={I(6, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACh, _3a1604AAACg, _3a1604AAACe, _3a1604AAACM, k(6, 5)], tags={I(6, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACj, _3a1604AAACi, _3a1604AAACg, _3a1604AAACO, k(6, 6)], tags={I(6, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACl, _3a1604AAACk, _3a1604AAACi, _3a1604AAACQ, k(6, 7)], tags={I(6, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACn, _3a1604AAACm, _3a1604AAACk, _3a1604AAACS, k(6, 8)], tags={I(6, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACp, _3a1604AAACo, _3a1604AAACm, _3a1604AAACU, k(6, 9)], tags={I(6, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACq, _3a1604AAACo, _3a1604AAACV, k(6, 10)], tags={I(6, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAACs, _3a1604AAACr, _3a1604AAACX, k(7, 0)], tags={I(7, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACu, _3a1604AAACt, _3a1604AAACr, _3a1604AAACZ, k(7, 1)], tags={I(7, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACw, _3a1604AAACv, _3a1604AAACt, _3a1604AAACb, k(7, 2)], tags={I(7, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAACy, _3a1604AAACx, _3a1604AAACv, _3a1604AAACd, k(7, 3)], tags={I(7, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADA, _3a1604AAACz, _3a1604AAACx, _3a1604AAACf, k(7, 4)], tags={I(7, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADC, _3a1604AAADB, _3a1604AAACz, _3a1604AAACh, k(7, 5)], tags={I(7, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADE, _3a1604AAADD, _3a1604AAADB, _3a1604AAACj, k(7, 6)], tags={I(7, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADG, _3a1604AAADF, _3a1604AAADD, _3a1604AAACl, k(7, 7)], tags={I(7, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADI, _3a1604AAADH, _3a1604AAADF, _3a1604AAACn, k(7, 8)], tags={I(7, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADK, _3a1604AAADJ, _3a1604AAADH, _3a1604AAACp, k(7, 9)], tags={I(7, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAADL, _3a1604AAADJ, _3a1604AAACq, k(7, 10)], tags={I(7, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAADN, _3a1604AAADM, _3a1604AAACs, k(8, 0)], tags={I(8, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADP, _3a1604AAADO, _3a1604AAADM, _3a1604AAACu, k(8, 1)], tags={I(8, 1)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADR, _3a1604AAADQ, _3a1604AAADO, _3a1604AAACw, k(8, 2)], tags={I(8, 2)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADT, _3a1604AAADS, _3a1604AAADQ, _3a1604AAACy, k(8, 3)], tags={I(8, 3)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADV, _3a1604AAADU, _3a1604AAADS, _3a1604AAADA, k(8, 4)], tags={I(8, 4)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADX, _3a1604AAADW, _3a1604AAADU, _3a1604AAADC, k(8, 5)], tags={I(8, 5)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADZ, _3a1604AAADY, _3a1604AAADW, _3a1604AAADE, k(8, 6)], tags={I(8, 6)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADb, _3a1604AAADa, _3a1604AAADY, _3a1604AAADG, k(8, 7)], tags={I(8, 7)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADd, _3a1604AAADc, _3a1604AAADa, _3a1604AAADI, k(8, 8)], tags={I(8, 8)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 1, 2), inds=[_3a1604AAADf, _3a1604AAADe, _3a1604AAADc, _3a1604AAADK, k(8, 9)], tags={I(8, 9)}),backend=numpy, dtype=float64, data=array([[[[[1., 0.]]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAADg, _3a1604AAADe, _3a1604AAADL, k(8, 10)], tags={I(8, 10)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])
Tensor(shape=(1, 1, 1, 2), inds=[_3a1604AAADi, _3a1604AAADh, _3a1604AAADN, k(9, 0)], tags={I(9, 0)}),backend=numpy, dtype=float64, data=array([[[[1., 0.]]]])

...

results = {}

# truncation options
max_bond = 16
cutoff = 1e-6

t = 0.0
dt = 0.04

U2 = ar.do("linalg.expm", -1j * dt * h2)
U1 = ar.do("linalg.expm", -1j * dt * h1)
mag_terms = {(site,): qu.pauli("Z") for site in [(Lx // 2, Ly // 2)]}

cluster_sets = {c: tuple(psi.gen_gloops_sites(c)) for c in [0, 4, 6]}
[len(cs) for cs in cluster_sets.values()]
[0, 100, 280]
def measure():
    if "times" in results and results["times"][-1] == t:
        # already measured this time
        return

    results.setdefault("times", []).append(t)

    for c, gloops in cluster_sets.items():
        results.setdefault(("norm", c), []).append(
            psi.norm_gloop_expand(
                gauges=gauges,
                gloops=gloops,
                optimize="random-greedy",
                autocomplete=True,
                autoreduce=True,
            ).real
        )

    # store some intermediate state for efficiency
    info = {}
    for c, gloops in cluster_sets.items():
        results.setdefault(("mag", c), []).append(
            psi.compute_local_expectation_gloop_expand(
                mag_terms,
                gauges=gauges,
                normalized="prod",
                gloops=gloops,
                optimize="random-greedy",
                autocomplete=True,
                info=info,
            ).real
            / len(mag_terms)
        )
for i in tqdm.trange(10):
    measure()

    for edge in qtn.tnag.tebd.edge_coloring(
        edges,
        "random_sequential",
        group=False,
    ):
        psi.gate_simple_(
            U2,
            where=edge,
            gauges=gauges,
            max_bond=max_bond,
            cutoff=cutoff,
            renorm=False,
        )

    for site in sites:
        psi.gate_simple_(U1, (site,), gauges=gauges)

    # ensure gauge is equilibrated
    psi.gauge_all_simple_(
        max_iterations=1000,
        tol=1e-6,
        gauges=gauges,
        progbar=False,
    )

    t += dt

measure()
100%|β–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆ| 10/10 [11:32<00:00, 69.24s/it]
psi
TensorNetworkGenVector(tensors=121, indices=341)
Tensor(shape=(16, 16, 2), inds=[_7322eeAAAAB, _7322eeAAAAA, k(0, 0)], tags={I(0, 0)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAD, _7322eeAAAAC, _7322eeAAAAA, k(0, 1)], tags={I(0, 1)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAF, _7322eeAAAAE, _7322eeAAAAC, k(0, 2)], tags={I(0, 2)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAH, _7322eeAAAAG, _7322eeAAAAE, k(0, 3)], tags={I(0, 3)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAJ, _7322eeAAAAI, _7322eeAAAAG, k(0, 4)], tags={I(0, 4)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAL, _7322eeAAAAK, _7322eeAAAAI, k(0, 5)], tags={I(0, 5)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAN, _7322eeAAAAM, _7322eeAAAAK, k(0, 6)], tags={I(0, 6)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAP, _7322eeAAAAO, _7322eeAAAAM, k(0, 7)], tags={I(0, 7)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAR, _7322eeAAAAQ, _7322eeAAAAO, k(0, 8)], tags={I(0, 8)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAT, _7322eeAAAAS, _7322eeAAAAQ, k(0, 9)], tags={I(0, 9)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 2), inds=[_7322eeAAAAU, _7322eeAAAAS, k(0, 10)], tags={I(0, 10)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAW, _7322eeAAAAV, _7322eeAAAAB, k(1, 0)], tags={I(1, 0)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAY, _7322eeAAAAX, _7322eeAAAAV, _7322eeAAAAD, k(1, 1)], tags={I(1, 1)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAa, _7322eeAAAAZ, _7322eeAAAAX, _7322eeAAAAF, k(1, 2)], tags={I(1, 2)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAc, _7322eeAAAAb, _7322eeAAAAZ, _7322eeAAAAH, k(1, 3)], tags={I(1, 3)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAe, _7322eeAAAAd, _7322eeAAAAb, _7322eeAAAAJ, k(1, 4)], tags={I(1, 4)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAg, _7322eeAAAAf, _7322eeAAAAd, _7322eeAAAAL, k(1, 5)], tags={I(1, 5)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAi, _7322eeAAAAh, _7322eeAAAAf, _7322eeAAAAN, k(1, 6)], tags={I(1, 6)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAk, _7322eeAAAAj, _7322eeAAAAh, _7322eeAAAAP, k(1, 7)], tags={I(1, 7)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAm, _7322eeAAAAl, _7322eeAAAAj, _7322eeAAAAR, k(1, 8)], tags={I(1, 8)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAo, _7322eeAAAAn, _7322eeAAAAl, _7322eeAAAAT, k(1, 9)], tags={I(1, 9)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAp, _7322eeAAAAn, _7322eeAAAAU, k(1, 10)], tags={I(1, 10)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAAAr, _7322eeAAAAq, _7322eeAAAAW, k(2, 0)], tags={I(2, 0)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAt, _7322eeAAAAs, _7322eeAAAAq, _7322eeAAAAY, k(2, 1)], tags={I(2, 1)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAv, _7322eeAAAAu, _7322eeAAAAs, _7322eeAAAAa, k(2, 2)], tags={I(2, 2)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAx, _7322eeAAAAw, _7322eeAAAAu, _7322eeAAAAc, k(2, 3)], tags={I(2, 3)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAAAz, _7322eeAAAAy, _7322eeAAAAw, _7322eeAAAAe, k(2, 4)], tags={I(2, 4)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABB, _7322eeAAABA, _7322eeAAAAy, _7322eeAAAAg, k(2, 5)], tags={I(2, 5)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABD, _7322eeAAABC, _7322eeAAABA, _7322eeAAAAi, k(2, 6)], tags={I(2, 6)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABF, _7322eeAAABE, _7322eeAAABC, _7322eeAAAAk, k(2, 7)], tags={I(2, 7)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABH, _7322eeAAABG, _7322eeAAABE, _7322eeAAAAm, k(2, 8)], tags={I(2, 8)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABJ, _7322eeAAABI, _7322eeAAABG, _7322eeAAAAo, k(2, 9)], tags={I(2, 9)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAABK, _7322eeAAABI, _7322eeAAAAp, k(2, 10)], tags={I(2, 10)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAABM, _7322eeAAABL, _7322eeAAAAr, k(3, 0)], tags={I(3, 0)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABO, _7322eeAAABN, _7322eeAAABL, _7322eeAAAAt, k(3, 1)], tags={I(3, 1)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABQ, _7322eeAAABP, _7322eeAAABN, _7322eeAAAAv, k(3, 2)], tags={I(3, 2)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABS, _7322eeAAABR, _7322eeAAABP, _7322eeAAAAx, k(3, 3)], tags={I(3, 3)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABU, _7322eeAAABT, _7322eeAAABR, _7322eeAAAAz, k(3, 4)], tags={I(3, 4)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABW, _7322eeAAABV, _7322eeAAABT, _7322eeAAABB, k(3, 5)], tags={I(3, 5)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABY, _7322eeAAABX, _7322eeAAABV, _7322eeAAABD, k(3, 6)], tags={I(3, 6)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABa, _7322eeAAABZ, _7322eeAAABX, _7322eeAAABF, k(3, 7)], tags={I(3, 7)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABc, _7322eeAAABb, _7322eeAAABZ, _7322eeAAABH, k(3, 8)], tags={I(3, 8)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABe, _7322eeAAABd, _7322eeAAABb, _7322eeAAABJ, k(3, 9)], tags={I(3, 9)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAABf, _7322eeAAABd, _7322eeAAABK, k(3, 10)], tags={I(3, 10)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAABh, _7322eeAAABg, _7322eeAAABM, k(4, 0)], tags={I(4, 0)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABj, _7322eeAAABi, _7322eeAAABg, _7322eeAAABO, k(4, 1)], tags={I(4, 1)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABl, _7322eeAAABk, _7322eeAAABi, _7322eeAAABQ, k(4, 2)], tags={I(4, 2)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABn, _7322eeAAABm, _7322eeAAABk, _7322eeAAABS, k(4, 3)], tags={I(4, 3)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABp, _7322eeAAABo, _7322eeAAABm, _7322eeAAABU, k(4, 4)], tags={I(4, 4)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABr, _7322eeAAABq, _7322eeAAABo, _7322eeAAABW, k(4, 5)], tags={I(4, 5)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABt, _7322eeAAABs, _7322eeAAABq, _7322eeAAABY, k(4, 6)], tags={I(4, 6)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABv, _7322eeAAABu, _7322eeAAABs, _7322eeAAABa, k(4, 7)], tags={I(4, 7)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABx, _7322eeAAABw, _7322eeAAABu, _7322eeAAABc, k(4, 8)], tags={I(4, 8)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAABz, _7322eeAAABy, _7322eeAAABw, _7322eeAAABe, k(4, 9)], tags={I(4, 9)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAACA, _7322eeAAABy, _7322eeAAABf, k(4, 10)], tags={I(4, 10)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAACC, _7322eeAAACB, _7322eeAAABh, k(5, 0)], tags={I(5, 0)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACE, _7322eeAAACD, _7322eeAAACB, _7322eeAAABj, k(5, 1)], tags={I(5, 1)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACG, _7322eeAAACF, _7322eeAAACD, _7322eeAAABl, k(5, 2)], tags={I(5, 2)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACI, _7322eeAAACH, _7322eeAAACF, _7322eeAAABn, k(5, 3)], tags={I(5, 3)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACK, _7322eeAAACJ, _7322eeAAACH, _7322eeAAABp, k(5, 4)], tags={I(5, 4)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACM, _7322eeAAACL, _7322eeAAACJ, _7322eeAAABr, k(5, 5)], tags={I(5, 5)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACO, _7322eeAAACN, _7322eeAAACL, _7322eeAAABt, k(5, 6)], tags={I(5, 6)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACQ, _7322eeAAACP, _7322eeAAACN, _7322eeAAABv, k(5, 7)], tags={I(5, 7)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACS, _7322eeAAACR, _7322eeAAACP, _7322eeAAABx, k(5, 8)], tags={I(5, 8)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACU, _7322eeAAACT, _7322eeAAACR, _7322eeAAABz, k(5, 9)], tags={I(5, 9)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAACV, _7322eeAAACT, _7322eeAAACA, k(5, 10)], tags={I(5, 10)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAACX, _7322eeAAACW, _7322eeAAACC, k(6, 0)], tags={I(6, 0)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACZ, _7322eeAAACY, _7322eeAAACW, _7322eeAAACE, k(6, 1)], tags={I(6, 1)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACb, _7322eeAAACa, _7322eeAAACY, _7322eeAAACG, k(6, 2)], tags={I(6, 2)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACd, _7322eeAAACc, _7322eeAAACa, _7322eeAAACI, k(6, 3)], tags={I(6, 3)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACf, _7322eeAAACe, _7322eeAAACc, _7322eeAAACK, k(6, 4)], tags={I(6, 4)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACh, _7322eeAAACg, _7322eeAAACe, _7322eeAAACM, k(6, 5)], tags={I(6, 5)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACj, _7322eeAAACi, _7322eeAAACg, _7322eeAAACO, k(6, 6)], tags={I(6, 6)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACl, _7322eeAAACk, _7322eeAAACi, _7322eeAAACQ, k(6, 7)], tags={I(6, 7)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACn, _7322eeAAACm, _7322eeAAACk, _7322eeAAACS, k(6, 8)], tags={I(6, 8)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACp, _7322eeAAACo, _7322eeAAACm, _7322eeAAACU, k(6, 9)], tags={I(6, 9)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAACq, _7322eeAAACo, _7322eeAAACV, k(6, 10)], tags={I(6, 10)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAACs, _7322eeAAACr, _7322eeAAACX, k(7, 0)], tags={I(7, 0)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACu, _7322eeAAACt, _7322eeAAACr, _7322eeAAACZ, k(7, 1)], tags={I(7, 1)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACw, _7322eeAAACv, _7322eeAAACt, _7322eeAAACb, k(7, 2)], tags={I(7, 2)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAACy, _7322eeAAACx, _7322eeAAACv, _7322eeAAACd, k(7, 3)], tags={I(7, 3)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADA, _7322eeAAACz, _7322eeAAACx, _7322eeAAACf, k(7, 4)], tags={I(7, 4)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADC, _7322eeAAADB, _7322eeAAACz, _7322eeAAACh, k(7, 5)], tags={I(7, 5)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADE, _7322eeAAADD, _7322eeAAADB, _7322eeAAACj, k(7, 6)], tags={I(7, 6)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADG, _7322eeAAADF, _7322eeAAADD, _7322eeAAACl, k(7, 7)], tags={I(7, 7)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADI, _7322eeAAADH, _7322eeAAADF, _7322eeAAACn, k(7, 8)], tags={I(7, 8)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADK, _7322eeAAADJ, _7322eeAAADH, _7322eeAAACp, k(7, 9)], tags={I(7, 9)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAADL, _7322eeAAADJ, _7322eeAAACq, k(7, 10)], tags={I(7, 10)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAADN, _7322eeAAADM, _7322eeAAACs, k(8, 0)], tags={I(8, 0)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADP, _7322eeAAADO, _7322eeAAADM, _7322eeAAACu, k(8, 1)], tags={I(8, 1)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADR, _7322eeAAADQ, _7322eeAAADO, _7322eeAAACw, k(8, 2)], tags={I(8, 2)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADT, _7322eeAAADS, _7322eeAAADQ, _7322eeAAACy, k(8, 3)], tags={I(8, 3)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADV, _7322eeAAADU, _7322eeAAADS, _7322eeAAADA, k(8, 4)], tags={I(8, 4)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADX, _7322eeAAADW, _7322eeAAADU, _7322eeAAADC, k(8, 5)], tags={I(8, 5)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADZ, _7322eeAAADY, _7322eeAAADW, _7322eeAAADE, k(8, 6)], tags={I(8, 6)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADb, _7322eeAAADa, _7322eeAAADY, _7322eeAAADG, k(8, 7)], tags={I(8, 7)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADd, _7322eeAAADc, _7322eeAAADa, _7322eeAAADI, k(8, 8)], tags={I(8, 8)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 16, 2), inds=[_7322eeAAADf, _7322eeAAADe, _7322eeAAADc, _7322eeAAADK, k(8, 9)], tags={I(8, 9)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAADg, _7322eeAAADe, _7322eeAAADL, k(8, 10)], tags={I(8, 10)}),backend=numpy, dtype=complex128, data=...
Tensor(shape=(16, 16, 16, 2), inds=[_7322eeAAADi, _7322eeAAADh, _7322eeAAADN, k(9, 0)], tags={I(9, 0)}),backend=numpy, dtype=complex128, data=...

...

for i, c in enumerate(cluster_sets):
    plt.plot(
        results["times"],
        results["norm", c],
        ".-",
        label=f"<psi|psi>C{c}",
        color=mpl.cm.Blues((i + 1) / len(cluster_sets)),
        linewidth=2 * (i + 1) / len(cluster_sets),
    )

for i, c in enumerate(cluster_sets):
    plt.plot(
        results["times"],
        results["mag", c],
        "|-",
        label=f"<M>C{c}",
        color=mpl.cm.Greens((i + 1) / len(cluster_sets)),
        linewidth=2 * (i + 1) / len(cluster_sets),
    )

plt.legend(bbox_to_anchor=(1, 1))
plt.show()
plt.close()
../_images/b2df5be7777e78258a2f100936929134db1804df9c2a82576b2c6dc5673f89bb.svg