python numpy.matmul 实现矩阵乘法

说明

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]]