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

TypeVar(T)