jax.lax.dot#
- jax.lax.dot(lhs, rhs, precision=None, preferred_element_type=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 (Array) – an array of dimension 1 or 2.
rhs (Array) – an array of dimension 1 or 2.
precision (PrecisionLike) – Optional. Either
None
, which means the default precision for the backend, aPrecision
enum value (Precision.DEFAULT
,Precision.HIGH
orPrecision.HIGHEST
) or a tuple of twoPrecision
enums indicating precision oflhs`
andrhs
.preferred_element_type (DTypeLike | None) – Optional. Either
None
, which means the default accumulation type for the input types, or a datatype, indicating to accumulate results to and return a result with that datatype.
- Return type:
Array
- Returns:
An array containing the product.