jax.lax.batch_matmul

Warning

This page was created from a pull request (#9655).

jax.lax.batch_matmul¶

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

Batch matrix multiplication.

Parameters

lhs (Any) –
rhs (Any) –
precision (Union[None, str, Any, Tuple[str, str], Tuple[Any, Any]]) –

Return type

previous

jax.lax.atan2

next

jax.lax.bessel_i0e