Skip to main content
Log in

Mojo module

math

Implements math methods that work on layout tensors.

Functions

  • exp: Computes element-wise exponential function.
  • max: Computes maximum reduction along specified axis.
  • outer_product_acc: Updates result tensor with the outer product of two vectors.
  • sum: Computes sum reduction along specified axis.