grape_functions.py 文件源码

def kron_all(op,num,op_2): 
    # returns an addition of sth like xii + ixi + iix for op =x and op_2 =i
    total = np.zeros([len(op)**num,len(op)**num])
    a=op
    for jj in range(num):
        if jj != 0:
            a = op_2
        else:
            a = op

        for ii in range(num-1):
            if (jj - ii) == 1:

                b = op
            else:
                b = op_2
            a = np.kron(a,b)
        total = total + a
    return a