Skip to main content
Log in

Mojo module

reduction

Implements SIMD reductions.

You can import these APIs from the algorithm package. For example:

from algorithm import map_reduce
from algorithm import map_reduce

Functions

  • all_true: Returns True if all the elements in a buffer are True and False otherwise.
  • any_true: Returns True if any the elements in a buffer are True and False otherwise.
  • argmax: Finds the indices of the maximum element along the specified axis.
  • argmin: Finds the indices of the minimum element along the specified axis.
  • cumsum: Computes the cumulative sum of all elements in a buffer. dst[i] = src[i] + src[i-1] + ... + src[0].
  • map_reduce: Stores the result of calling input_gen_fn in dst and simultaneously reduce the result using a custom reduction function.
  • max: Computes the max element in a buffer.
  • mean: Computes the mean value of the elements in a buffer.
  • min: Computes the min element in a buffer.
  • none_true: Returns True if none of the elements in a buffer are True and False otherwise.
  • product: Computes the product of the buffer elements.
  • reduce: Computes a custom reduction of buffer elements.
  • reduce_boolean: Computes a bool reduction of buffer elements. The reduction will early exit if the continue_fn returns False.
  • sum: Computes the sum of buffer elements.
  • variance: Given a mean, computes the variance of elements in a buffer.

Was this page helpful?