Advanced Computing Platform for Theoretical Physics

Commit 8b448682 authored by Lei Wang's avatar Lei Wang
Browse files

double layer

parent cb8d30a0
......@@ -75,8 +75,9 @@ class Ising(torch.nn.Module):
lnZ += 2**(-n)*torch.log(f)
T = T / f
T = torch.einsum('xabcdx->abcd', (T)) #trace
#T = torch.einsum('xabcdx->abcd', (T)) #trace
#T = symmetrize(T)
T = torch.einsum('xabcdy,xefghy->aebfcgdh', (T, T)).contiguous().view(self.D**2, self.D**2, self.D**2, self.D**2)
lnZ += ctmrg(T, self.chi, self.nctmrg)/2.0**self.nxtrg
return lnZ
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment