jaxopt.tree_util.tree_mul

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

Tree multiplication.

Parameters
  • tree (Any) –

  • rest (Any) –

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

Return type

Any