jax.Array.dot

Contents

jax.Array.dot#

abstract Array.dot(b, *, precision=None, preferred_element_type=None)[source]#

Compute the dot product of two arrays.

Refer to jax.numpy.dot() for the full documentation.

Parameters:
  • self (Array)

  • b (ArrayLike)

  • precision (lax_internal.PrecisionLike)

  • preferred_element_type (DTypeLike | None)

Return type:

Array