jax.lax.reduce¶

jax.lax.reduce(operand, init_value, computation, dimensions)[source]¶

Wraps XLA’s Reduce operator.

Parameters
Return type

Any