numpy.linalg.matmul#

linalg.matmul(x1, x2, /)[源代码]#

计算矩阵乘积。

此函数与 Array API 兼容，与 numpy.matmul 不同。

参数:
x1类数组

第一个输入数组。

x2类数组

第二个输入数组。

返回:
outndarray

输入的矩阵乘积。仅当 x1 和 x2 都是一维向量时，结果才是一个标量。

抛出:
ValueError

如果 x1 的最后一个维度与 x2 的倒数第二个维度大小不同。

如果传入标量值。

另请参阅

numpy.matmul

示例

对于二维数组，它是矩阵乘积

>>> a = np.array([[1, 0],
...               [0, 1]])
>>> b = np.array([[4, 1],
...               [2, 2]])
>>> np.linalg.matmul(a, b)
array([[4, 1],
       [2, 2]])

对于二维和一维混合的情况，结果与通常相同。

>>> a = np.array([[1, 0],
...               [0, 1]])
>>> b = np.array([1, 2])
>>> np.linalg.matmul(a, b)
array([1, 2])
>>> np.linalg.matmul(b, a)
array([1, 2])

对于数组堆栈，广播是惯例

>>> a = np.arange(2 * 2 * 4).reshape((2, 2, 4))
>>> b = np.arange(2 * 2 * 4).reshape((2, 4, 2))
>>> np.linalg.matmul(a,b).shape
(2, 2, 2)
>>> np.linalg.matmul(a, b)[0, 1, 1]
98
>>> sum(a[0, 1, :] * b[0 , :, 1])
98

向量与向量的乘积返回标量内积，但两个参数均未进行复共轭。

>>> np.linalg.matmul([2j, 3j], [2j, 3j])
(-13+0j)

标量乘法会引发错误。

>>> np.linalg.matmul([1,2], 3)
Traceback (most recent call last):
...
ValueError: matmul: Input operand 1 does not have enough dimensions ...
本页内容