numpy.matmul() 函数

numpy.matmul() 函数返回两个数组的矩阵乘积。虽然它返回二维数组的普通乘积，但如果任一参数的维度大于 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]]

# 二维与一维混合
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]

# 一个数组，维度 > 2
import numpy.matlib
import numpy as np

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

将产生以下输出 -

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