ott.geometry.regularizers.STVS#
- class ott.geometry.regularizers.STVS(gamma=1.0)[source]#
Soft thresholding operator with vanishing shrinkage regularizer [Schreck et al., 2016].
The operator is defined as:
\[\gamma^2 \mathbf{1}_d^T \left(\sigma(x) - \frac{1}{2} \exp\left(-2\sigma(x)\right) + \frac{1}{2}\right)\]where \(\sigma(x) := \text{asinh}\left(\frac{x}{2\gamma}\right)\).
- Parameters:
gamma (
float
) – Strength of the regularization.
Methods
moreau_envelope
(x[, tau])Moreau Envelope.
prox
(v[, tau])Proximal operator.
prox_dual
(v[, tau])Proximal operator of the convex conjugate.