@@ -18,7 +18,7 @@ def matvec(self, v):
1818 lbl = v .labels ()
1919 self .anet .PutUniTensor ("psi" ,v )
2020 out = self .anet .Launch ()
21- out .relabels_ (lbl )
21+ out .relabel_ (lbl )
2222 return out
2323
2424 def optimize_psi (psi , functArgs , maxit = 2 , krydim = 4 ):
@@ -61,7 +61,7 @@ def optimize_psi(psi, functArgs, maxit=2, krydim=4):
6161 lbls = [] # List for storing the MPS labels
6262 A = [None for i in range (Nsites )]
6363 A [0 ] = cytnx .UniTensor (cytnx .random .normal ([1 , d , min (chi , d )], 0. , 1. ), rowrank = 2 )
64- A [0 ].relabels_ (["0" ,"1" ,"2" ])
64+ A [0 ].relabel_ (["0" ,"1" ,"2" ])
6565 lbls .append (["0" ,"1" ,"2" ]) # store the labels for later convinience.
6666
6767 for k in range (1 ,Nsites ):
@@ -70,7 +70,7 @@ def optimize_psi(psi, functArgs, maxit=2, krydim=4):
7070 A [k ] = cytnx .UniTensor (cytnx .random .normal ([dim1 , dim2 , dim3 ],0. ,1. ), rowrank = 2 )
7171
7272 lbl = [str (2 * k ),str (2 * k + 1 ),str (2 * k + 2 )]
73- A [k ].relabels_ (lbl )
73+ A [k ].relabel_ (lbl )
7474 lbls .append (lbl ) # store the labels for later convinience.
7575
7676 LR = [None for i in range (Nsites + 1 )]
@@ -96,11 +96,11 @@ def optimize_psi(psi, functArgs, maxit=2, krydim=4):
9696 LR [p + 1 ] = anet .Launch ()
9797
9898 # Recover the original MPS labels
99- A [p ].relabels_ (lbls [p ])
100- A [p + 1 ].relabels_ (lbls [p + 1 ])
99+ A [p ].relabel_ (lbls [p ])
100+ A [p + 1 ].relabel_ (lbls [p + 1 ])
101101
102102 _ ,A [- 1 ] = cytnx .linalg .Gesvd (A [- 1 ],is_U = True ,is_vT = False ) ## last one.
103- A [- 1 ].relabels_ (lbls [- 1 ]) # Recover the original MPS labels
103+ A [- 1 ].relabel_ (lbls [- 1 ]) # Recover the original MPS labels
104104
105105 Ekeep = []
106106 for k in range (1 , numsweeps + 1 ):
@@ -116,12 +116,12 @@ def optimize_psi(psi, functArgs, maxit=2, krydim=4):
116116
117117 psi .set_rowrank_ (2 ) # maintain rowrank to perform the svd
118118 s ,A [p ],A [p + 1 ] = cytnx .linalg .Svd_truncate (psi ,new_dim )
119- A [p + 1 ].relabels_ (lbls [p + 1 ]); # set the label back to be consistent
119+ A [p + 1 ].relabel_ (lbls [p + 1 ]); # set the label back to be consistent
120120
121121 s = s / s .Norm ().item () # normalize s
122122
123123 A [p ] = cytnx .Contract (A [p ],s ) # absorb s into next neighbor
124- A [p ].relabels_ (lbls [p ]); # set the label back to be consistent
124+ A [p ].relabel_ (lbls [p ]); # set the label back to be consistent
125125
126126 # update LR from right to left:
127127 anet = cytnx .Network ()
@@ -138,7 +138,7 @@ def optimize_psi(psi, functArgs, maxit=2, krydim=4):
138138
139139 A [0 ].set_rowrank_ (1 )
140140 _ ,A [0 ] = cytnx .linalg .Gesvd (A [0 ],is_U = False , is_vT = True )
141- A [0 ].relabels_ (lbls [0 ]); #set the label back to be consistent
141+ A [0 ].relabel_ (lbls [0 ]); #set the label back to be consistent
142142
143143 for p in range (Nsites - 1 ):
144144 dim_l = A [p ].shape ()[0 ]
@@ -151,12 +151,12 @@ def optimize_psi(psi, functArgs, maxit=2, krydim=4):
151151
152152 psi .set_rowrank_ (2 ) # maintain rowrank to perform the svd
153153 s ,A [p ],A [p + 1 ] = cytnx .linalg .Svd_truncate (psi ,new_dim )
154- A [p ].relabels_ (lbls [p ]); #set the label back to be consistent
154+ A [p ].relabel_ (lbls [p ]); #set the label back to be consistent
155155
156156 s = s / s .Norm ().item () # normalize s
157157
158158 A [p + 1 ] = cytnx .Contract (s ,A [p + 1 ]) ## absorb s into next neighbor.
159- A [p + 1 ].relabels_ (lbls [p + 1 ]); #set the label back to be consistent
159+ A [p + 1 ].relabel_ (lbls [p + 1 ]); #set the label back to be consistent
160160
161161 # update LR from left to right:
162162 anet = cytnx .Network ()
@@ -174,7 +174,7 @@ def optimize_psi(psi, functArgs, maxit=2, krydim=4):
174174
175175 A [- 1 ].set_rowrank_ (2 )
176176 _ ,A [- 1 ] = cytnx .linalg .Gesvd (A [- 1 ],is_U = True ,is_vT = False ) ## last one.
177- A [- 1 ].relabels_ (lbls [- 1 ]); #set the label back to be consistent
177+ A [- 1 ].relabel_ (lbls [- 1 ]); #set the label back to be consistent
178178 return Ekeep
179179
180180if __name__ == '__main__' :
0 commit comments