jaxopt.projection.projection_l2_sphere
- jaxopt.projection.projection_l2_sphere(x, value=1.0)[source]
Projection onto the l2 sphere:
\[\underset{y}{\text{argmin}} ~ ||x - y||_2^2 \quad \textrm{subject to} \quad ||y||_2 = \text{value}\]- Parameters
x (
Any) – pytree to project.value (
float) – radius of the sphere.
- Return type
Any- Returns
output pytree, with the same structure as
x.