For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo function
batch_matmul_shape
def batch_matmul_shape(a: T, b: T) -> IndexList[T.rank]
Computes the output shape for the mo.batch_matmul graph op.
Args:
- a (
T): Left-hand batched input tensor. - b (
T): Right-hand batched input tensor.
Returns:
IndexList[T.rank]: The output shape of the batched matmul.