jax.numpy.vdot

jax.numpy.vdot(a, b, *, precision=None, preferred_element_type=None)

執行兩個 1D 向量的共軛乘法。

JAX 版本的 numpy.vdot()。

參數:

• a (ArrayLike) – 第一個輸入陣列，如果不是 1D 將被展平。

• b (ArrayLike) – 第二個輸入陣列，如果不是 1D 將被展平。必須具有 a.size == b.size。

• precision (PrecisionLike) – None (預設)，表示後端的預設精度，Precision 列舉值 (Precision.DEFAULT, Precision.HIGH 或 Precision.HIGHEST) 或兩個此類值的元組，表示 a 和 b 的精度。

• preferred_element_type (DTypeLike | None) – None (預設)，表示輸入類型的預設累積類型，或是一種資料類型，表示累積結果並傳回具有該資料類型的結果。

傳回:

純量陣列 (形狀 ())，包含輸入的共軛向量積。

傳回類型:

Array

另請參閱

• jax.numpy.vecdot()：批次向量積。

• jax.numpy.matmul()：一般矩陣乘法。

• jax.lax.dot_general()：一般 N 維批次點積。

範例

>>> x = jnp.array([1j, 2j, 3j])
>>> y = jnp.array([1., 2., 3.])
>>> jnp.vdot(x, y)
Array(0.-14.j, dtype=complex64)

請注意此函式與 dot() 之間的差異，後者在複數情況下不會共軛第一個輸入

>>> jnp.dot(x, y)
Array(0.+14.j, dtype=complex64)