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]]
示例
# 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]]]