Skip to main content

Mojo module

math

Implements math methods that work on layout tensors.

Functions

  • max: Computes maximum reduction along specified axis.
  • mean: Computes the mean value of the elements in a buffer.
  • outer_product_acc: Updates result tensor with the outer product of two vectors.
  • sum: Computes sum reduction along specified axis.
  • variance: Computes the variance value of the elements in a buffer.

Was this page helpful?