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