动态规划--矩阵链乘法

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