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