Java教程

[算法导论] 矩阵乘法

本文主要是介绍[算法导论] 矩阵乘法，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！ 
import numpy as np


def matmul(a, b):
    return np.matmul(a, b)


def manual_matmul(a, b):
    C = []
    for x in range(len(a)):
        tp = []
        for y in range(len(b[0])):
            tp.append(sum(a[x][k]*b[k][y] for k in range(len(b))))
        C.append(tp)
    return np.array(C)


if __name__ == '__main__':
    # A=[],B=[] 这是错的
    A = []
    B = []
    na, ma = map(int, input().split())
    # 按行存储就可以
    for i in range(na):
        tmp = []
        tmp = list(map(int, input().split()))
        A.append(tmp)
    nb, mb = map(int, input().split())
    for i in range(nb):
        tmp = []
        tmp = list(map(int, input().split()))
        B.append(tmp)

    print("熊熊")
    print(A)
    print(B)
    #print(matmul(A, B))
    print(manual_matmul(A, B))

这篇关于[算法导论] 矩阵乘法的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！
您可能喜欢
栏目导航