结论:
A ∗ B = ( i , m ) ∗ ( m , j ) A*B=(i, m)*(m, j) AB=(i,m)(m,j)的计算复杂度为 O ( i ∗ j ∗ m ) O(i*j*m) O(ijm)
怎么来的:

输出矩阵C为 ( i , j ) (i, j) (i,j),遍历输出矩阵,C矩阵中的每个元素由A,B矩阵中对应的m对数字相乘相加而来,遍历m对数做相乘相加的复杂度为 O ( m ) O(m) O(m),所以总共三轮循环,复杂度为 O ( i ∗ j ∗ m ) O(i*j*m) O(ijm)

实际实现:

a = [[1, 2, 3],
     [4, 5, 6]]

b = [[1, 4],
     [2,5],
     [3,6]]

c = [[0, 0],
     [0, 0]]#初始化为全0
for i in range(2):
    for j in range(2):
        for m in range(3):
             c[i, j] = c[i, j] + a[i, m]*b[m, j]
Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐