Warning

This page was created from a pull request.

jax.lax.batch_matmul

jax.lax.batch_matmul(lhs, rhs, precision=None)

Batch matrix multiplication.

Parameters

• lhs (Any) –

• rhs (Any) –

• precision (Union[None, Any, Tuple[Any, Any]]) –

Return type

Any