tfmri.math.shrinkage

shrinkage(x, threshold, name=None)[source]

Shrinkage operator.

In the context of proximal gradient methods, this function is the proximal operator of \(f = \frac{1}{2}{\left\| x \right\|}_{2}^{2}\).

Parameters

  • x – A Tensor of shape [..., n].

  • threshold – A float.

  • name – A name for this operation (optional).

Returns

A Tensor of shape [..., n] and same dtype as x.