TRG/Combiner.py at main · pcchen/TRG · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import cytnx
import numpy as np
import itertools

# bd = cytnx.Bond(2)

# T = cytnx.UniTensor([bd,bd,bd,bd]).set_rowrank(2).set_name('T').relabel(['a','b','c','d'])
# T.print_diagram()
# print(T[0,0,0,0].item())

# for i in range(2):
#     for j in range(2):
#         T[i,j,i,j] = 1

# print(T)
# T.combineBonds(['a','c'])
# T.print_diagram()
# exit()
# # T.bonds()[2].combine
# T.combineBonds(['c','d'])
# T.print_diagram()
# # print(T)
# for i in range(2):
#     for j in range(2):
#         for k in range(4):
#             print(i,j,k,T[i,j,k].item())


def combiner(T, bds_in, bds_out):
    """
    bds_in: list of bonds to be combined
    bds_out: list of bonds to be output
    """
    for bd in bds_in:
        print(T.bond(bd))
    # comb = cytnx.UniTensor.(bds_in+bds_in)
    # comb.print_diagram()
    # return comb


T = cytnx.UniTensor.ones([2,2,2,2]).set_rowrank(2).set_name('T').relabel(['a','b','c','d'])
T.print_diagram()

comb =combiner(T, ['c','d'],['e'])
# comb.print_diagram()


exit()