ott.geometry.costs.ElasticSTVS.prox_reg

ElasticSTVS.prox_reg(z, tau=1.0)

Proximal operator of reg().

Parameters:

• z (Array) – Array of shape [d,].

• tau (float) – Positive weight.

Return type:

Array

Returns:

The prox of z.