JAXopt
Documentation
Basics
Unconstrained optimization
Constrained optimization
Quadratic programming
Non-smooth optimization
Stochastic optimization
Root finding
Fixed point resolution
Nonlinear least squares
Linear system solving
Implicit differentiation
Loss and objective functions
Line search
Perturbed optimization
API
API at a glance
Optimization
Linear system solving
Nonlinear least squares
Root finding
Fixed point resolution
Implicit differentiation
Line search
Perturbed optimizers
Isotonic regression
Tree utilities
jaxopt.tree_util.tree_add
jaxopt.tree_util.tree_sub
jaxopt.tree_util.tree_mul
jaxopt.tree_util.tree_div
jaxopt.tree_util.tree_scalar_mul
jaxopt.tree_util.tree_add_scalar_mul
tree_add_scalar_mul()
jaxopt.tree_util.tree_vdot
jaxopt.tree_util.tree_sum
jaxopt.tree_util.tree_l2_norm
jaxopt.tree_util.tree_zeros_like
Examples
Notebook gallery
Example gallery
About
Authors
Changelog
Source code
Issue tracker
Development
JAXopt
»
API at a glance
»
jaxopt.tree_util.tree_add_scalar_mul
View page source
jaxopt.tree_util.tree_add_scalar_mul
jaxopt.tree_util.
tree_add_scalar_mul
(
tree_x
,
scalar
,
tree_y
)
[source]
Compute tree_x + scalar * tree_y.