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!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit