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!