makeNet.py 文件源码

def combineTransMats(mats_trans,states_pho):
    '''
    combine multi transition matrix into one matrix
    :param mats_trans:
    :param states_pho:
    :return:
    '''
    mat_trans_comb = block_diag(*mats_trans)
    state_pho_comb = sum(states_pho, [])
    index_start = [0]
    index_end = [mats_trans[0].shape[0]-1]
    for ii in range(1,len(mats_trans)):
        index_start.append(index_end[-1]+1)
        index_end.append(index_end[-1]+mats_trans[ii].shape[0])
    return mat_trans_comb,state_pho_comb,index_start,index_end