jax.tree_util.tree_reduce

jax.tree_util.tree_reduce(function, tree, initializer=<object object>, is_leaf=None)

Parameters

• function (Callable[[T, Any], T]) –

• tree (Any) –

• initializer (Any) –

• is_leaf (Callable[[Any], bool] | None) –

Return type

T