当前位置: 首页 > 图灵资讯 > 行业资讯> python numpy.matmul实现矩阵相乘

python numpy.matmul实现矩阵相乘

来源:图灵python
时间: 2024-07-04 14:31:11

说明

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

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

实例

importnumpy.matlib
importnumpyasnp

a=[1,0],[0,1]
b=[4,1],[2,2]]
print(np.matmul(a,b))
输出结果为:

[[41]
[22]]

以上是python numpy.如何实现matmul矩阵相乘,希望对大家有所帮助。更多Python学习指导:python基础教程