Mojo module

linalg

Ops that perform linear algebra.

Functions

• ​band_part: Masks out everything except a diagonal band of an input matrix.
• ​layer_norm: Performs layer normalization.
• ​matmul: Computes the matrix multiplication of two symbolic tensors.
• ​outer: Computes the outer product of two symbolic vectors.
• ​tile: Returns a new Tensor as the result of copying the input tensor N_i times on each dimension, where N_i = tiles[i].

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!
If you'd like to share more information, please report an issue on GitHub

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other