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 (Optional[Any]) – Optional. Either None, which means the default precision for the backend, or a lax.Precision enum value (Precision.DEFAULT, Precision.HIGH or Precision.HIGHEST).

Return type

Any

Returns

An array containing the product.