动态规划--矩阵链乘法

来源:互联网 发布:淘宝海外代购是真的吗 编辑:程序博客网 时间:2024/06/05 00:46

1.两个矩阵乘法

def matrix_multipy(A, B):    '''    乘法得到的是一个[A.rows,B.cols]的矩阵, 相当于A.rows个向量的B.cols次的向量线性加权    '''    if not A.shape[1]==B.shape[0]:        #A组中向量的维度与B组中向量的维度一致        print("error!")    else:        #将每一个元素都初始化为0        C = np.zeros([A.shape[0], B.shape[1]])        for i in range(A.shape[0]):            for j in range(B.shape[1]):                for k in range(A.shape[1]):                    C[i][j] += A[i][k]*B[k][j]    return C
原创粉丝点击