Skip to main content
Ctrl+K

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)

previous

jax.tree_util.tree_map_with_path

next

jax.tree_util.tree_structure