python numpy.matmul实现矩阵相乘-Python基础教程-Python学习网

python numpy.matmul实现矩阵相乘

小妮浅浅2021-08-02 15:52:04原创

说明

1、该函数返回两个数组的矩阵乘积。虽然返回二维数组的正常乘积，但如果任何参数的维数大于2，则视为存在于最后两个索引的矩阵栈中并进行相应的广播。

2、如果任何参数都是一维数组，则通过在其维度上添加1来将其提升为矩阵，并在乘法后被移除。

实例

import numpy.matlib
import numpy as np
  
a = [[1,0],[0,1]]
b = [[4,1],[2,2]]
print (np.matmul(a,b))
输出结果为：
  
[[4  1]
 [2  2]]

以上就是python numpy.matmul实现矩阵相乘的方法，希望对大家有所帮助。更多Python学习指路：python基础教程

专题推荐：python numpymatmul

python numpy.matmul实现矩阵相乘