jax.tree_util.tree_reduce# jax.tree_util.tree_reduce(function, tree, initializer=<object object>, is_leaf=None)[source]# Parameters: function (Callable[[TypeVar(T), Any], TypeVar(T)]) – tree (Any) – initializer (Any) – is_leaf (Optional[Callable[[Any], bool]]) – Return type: TypeVar(T)