jaxopt.tree_util.tree_add

jaxopt.tree_util.tree_add(tree, *rest, is_leaf=None)

Tree addition.

Parameters
  • tree (Any) –

  • rest (Any) –

  • is_leaf (Optional[Callable[[Any], bool]]) –

Return type

Any