ott.geometry.regularizers.Regularization#
- class ott.geometry.regularizers.Regularization(f, a=None, rho=1.0)[source]#
Regularization operator \(f\left(x\right) + \frac{\rho}{2}\|x - a\|_2^2\).
- Parameters:
f (
ProximalOperator
) – Function \(f\).a (
Optional
[Array
]) – Offset \(a\). IfNone
, use array of 0s.rho (
float
) – Scaling factor.
Methods
moreau_envelope
(x[, tau])Moreau Envelope.
prox
(v[, tau])Proximal operator.
prox_dual
(v[, tau])Proximal operator of the convex conjugate.