jaxopt.tree_util.tree_sub

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

Tree subtraction.

Parameters

• tree (Any) –

• rest (Any) –

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

Return type

Any