Ctrl+K

jax.numpy.vecdot

Contents

jax.numpy.vecdot#

jax.numpy.vecdot(x1, x2, /, *, axis=-1, precision=None, preferred_element_type=None)[source]#

In addition to the original NumPy arguments listed below, also supports precision for extra control over matrix-multiplication precision on supported devices. precision may be set to None, which means default precision for the backend, a Precision enum value (Precision.DEFAULT, Precision.HIGH or Precision.HIGHEST) or a tuple of two Precision enums indicating separate precision for each argument.

Parameters:

  • x1 (ArrayLike)

  • x2 (ArrayLike)

  • axis (int)

  • precision (PrecisionLike)

  • preferred_element_type (DTypeLike | None)

Return type:

Array

Contents