jax.tree_util.tree_reduce

Warning

This page was created from a pull request (#9655).

jax.tree_util.tree_reduce

jax.tree_util.tree_reduce(function: Callable[[jax._src.tree_util.T, Any], jax._src.tree_util.T], tree: Any) → jax._src.tree_util.T[source]
jax.tree_util.tree_reduce(function: Callable[[jax._src.tree_util.T, Any], jax._src.tree_util.T], tree: Any, initializer: jax._src.tree_util.T) → jax._src.tree_util.T
Parameters
Return type

~T

previous

jax.tree_util.tree_map

next

jax.tree_util.tree_structure