ott.geometry.regularizers.STVS.prox_dual

Contents

ott.geometry.regularizers.STVS.prox_dual#

STVS.prox_dual(v, tau=1.0)#

Proximal operator of the convex conjugate.

Uses Moreau’s decomposition:

\[v = \prox_{\tau f} \left(v\right) + \tau \prox_{\frac{1}{\tau} f^*} \left(\frac{v}{\tau}\right)\]
Parameters:
  • v (Array) – Array of shape [d,].

  • tau (float) – Positive weight.

Return type:

Array

Returns:

The prox dual of v.