- jax.lax.reduce(operands, init_values, computation, dimensions)¶
Wraps XLA’s Reduce operator.
computationtogether must form a monoid for correctness. That is
init_valuesmust be an identity of
computationmust be associative. XLA may exploit both of these properties during code generation; if either is violated the result is undefined.