NumPy 中的 matmul函数

首页 / Numpy入门教程 / NumPy 中的 matmul函数

numpy.matmul()函数返回两个数组的矩阵乘积,虽然它为二维数组返回正常乘积,但如果任一参数的维数> 2,则将其视为驻留在最后两个索引中的矩阵堆栈,并进行相应广播。

另一方面,如果任一自变量是一维数组,则通过在其维上附加1来将其提升为矩阵,然后在乘法后将其删除。

# For 2-D array, it is matrix multiplication 
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]] 
# 2-D mixed with 1-D 
import numpy.matlib 
import numpy as np 

a = [[1,0],[0,1]] 
b = [1,2] 
print np.matmul(a,b) 
print np.matmul(b,a)

它将产生以下输出-

无涯教程网

[1  2] 
[1  2] 
# one array having dimensions > 2 
import numpy.matlib 
import numpy as np 

a = np.arange(8).reshape(2,2,2) 
b = np.arange(4).reshape(2,2) 
print np.matmul(a,b)

它将产生以下输出-

无涯教程网

[[[2   3] 
   [6   11]] 
  [[10  19] 
   [14  27]]]

祝学习愉快!(内容编辑有误?请选中要编辑内容 -> 右键 -> 修改 -> 提交!)

技术教程推荐

iOS开发高手课 -〔戴铭〕

研发效率破局之道 -〔葛俊〕

Netty源码剖析与实战 -〔傅健〕

说透敏捷 -〔宋宁〕

TensorFlow 2项目进阶实战 -〔彭靖田〕

全链路压测实战30讲 -〔高楼〕

深入C语言和程序运行原理 -〔于航〕

快速上手C++数据结构与算法 -〔王健伟〕

LangChain 实战课 -〔黄佳〕

好记忆不如烂笔头。留下您的足迹吧 :)