Python numpy vdot用法及代码示例

本文简要介绍 python 语言中 numpy.vdot 的用法。

用法:

numpy.vdot(a, b, /)

返回两个向量的点积。

vdot(a, b) 函数处理复数的方式与 dot(a, b) 不同。如果第一个参数是复数，则第一个参数的复共轭用于计算点积。

注意vdot处理多维数组的方式不同于numpy.dot: 确实不是执行矩阵乘积，但首先将输入参数展平为一维向量。因此，它应该只用于向量。

参数：

a： array_like

如果 a 是复数，则在计算点积之前取复共轭。

b： array_like

点积的第二个参数。

返回：

output： ndarray

a 和 b 的点积。根据 a 和 b 的类型，可以是 int、float 或 complex。

例子：

>>> a = np.array([1+2j,3+4j])
>>> b = np.array([5+6j,7+8j])
>>> np.vdot(a, b)
(70-8j)
>>> np.vdot(b, a)
(70+8j)

请注意，高维数组是扁平的！

>>> a = np.array([[1, 4], [5, 6]])
>>> b = np.array([[4, 1], [2, 2]])
>>> np.vdot(a, b)
30
>>> np.vdot(b, a)
30
>>> 1*4 + 4*1 + 5*2 + 6*2
30