jax.lax.batch_matmul# jax.lax.batch_matmul(lhs, rhs, precision=None)[source]# Batch matrix multiplication. Parameters: lhs (Array) – rhs (Array) – precision (Union[str, Precision, tuple[str, str], tuple[Precision, Precision], None]) – Return type: Array
