用python实现矩阵相乘——目的是通过详细实现过程了解矩阵相乘步骤

代码:

#! /usr/bin/env python
#coding=utf-8

import numpy as np
def MatrixMultiply(a, b):
    n11,n12=np.shape(a)
    n21,n22=np.shape(b)
    n=max(n11,n22)
    c=np.zeros([n11,n22],dtype=int)
    for i in range(0, n11):
        for j in range(0, n22):
            #c[i][j]=0
            for k in range(0, n21):
                c[i][j]=c[i][j]+a[i][k]*b[k][j]
    return c

if __name__ == '__main__':
    a=[[1,1],[2,3]]
    b=[[1,0,0,1],[2,1,2,4]]
    c=MatrixMultiply(a, b)
    print(c)

注意:

  • 矩阵相乘的维度要求

直接调用内置函数实现的资料推荐:

猜你喜欢

转载自blog.csdn.net/qq_39059714/article/details/89360253