jaxopt.projection.projection_l2_sphere

jaxopt.projection.projection_l2_sphere(x, value=1.0)

Projection onto the l2 sphere:

$$\underset{y}{\text{argmin}}||x-y|{|}_2^2\text{subject to}||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.