Warning

This page was created from a pull request.

jax.lax.dot¶

jax.lax.dot(lhs, rhs, precision=None)[source]¶

Vector/vector, matrix/vector, and matrix/matrix multiplication.

Wraps XLA’s Dot operator.

For more general contraction, see the dot_general operator.

Parameters

  • lhs (Any) – an array of rank 1 or 2.

  • rhs (Any) – an array of rank 1 or 2.

  • precision (Union[None, Any, Tuple[Any, Any]]) – Optional. Either None, which means the default precision for the backend, a lax.Precision enum value (Precision.DEFAULT, Precision.HIGH or Precision.HIGHEST) or a tuple of two lax.Precision enums indicating precision of lhs` and rhs.

Return type

Any

Returns

An array containing the product.