jax.numpy.vdot

目录

jax.numpy.vdot#

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

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

JAX 实现的 numpy.vdot()。

参数:

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

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

  • precision (PrecisionLike) – 可以是 None``（默认），这意味着后端的默认精度，一个 :class:`~jax.lax.Precision` 枚举值（``Precision.DEFAULT、Precision.HIGH 或 Precision.HIGHEST），或者是一个包含两个此类值的元组，指示 a 和 b 的精度。

  • preferred_element_type (DTypeLike | None) – ``None``（默认），这意味着输入类型的默认累积类型，或者是一个数据类型，指示将结果累积到并返回该数据类型的结果。

返回:

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

返回类型:

Array

参见

示例

>>> 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)
目录