jax.numpy.vdot

内容

jax.numpy.vdot#

jax.numpy.vdot(a, b, *, precision=None, preferred_element_type=None)[source]#

执行两个一维向量的共轭乘法。

numpy.vdot() 的 JAX 实现。

参数：

  • a (ArrayLike) – 第一个输入数组，如果不是一维数组，它将被展平。

  • b (ArrayLike) – 第二个输入数组，如果不是一维数组，它将被展平。必须有 a.size == b.size。

  • precision (PrecisionLike) – None（默认值），表示后端的默认精度，一个 Precision 枚举值（Precision.DEFAULT、Precision.HIGH 或 Precision.HIGHEST）或两个这样的值的元组，指示 a 和 b 的精度。

  • preferred_element_type (DTypeLike | None) – None（默认值），表示输入类型的默认累积类型，或一种数据类型，指示将结果累积到并返回具有该数据类型的结果。

返回值：

包含输入的共轭向量积的标量数组（形状 ()）。

返回类型：

数组

另请参阅

示例

>>> 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)
内容