Warning

This page was created from a pull request.

jax.lax.reduce

jax.lax.reduce(operands, init_values, computation, dimensions)

Wraps XLA’s Reduce operator.

Parameters

• operands (Any) –

• init_values (Any) –

• computation (Callable) –

• dimensions (Sequence[int]) –

Return type

Any