Warning: This project is deprecated. TensorFlow Addons has stopped development, The project will only be providing minimal maintenance releases until May 2024. See the full announcement here or on github.


Soft shrink function.

Computes soft shrink function:

\[ \mathrm{softshrink}(x) = \begin{cases} x - \mathrm{lower} & \text{if } x < \mathrm{lower} \\ x - \mathrm{upper} & \text{if } x > \mathrm{upper} \\ 0 & \text{otherwise} \end{cases}. \]


x = tf.constant([-1.0, 0.0, 1.0])
<tf.Tensor: shape=(3,), dtype=float32, numpy=array([-0.5,  0. ,  0.5], dtype=float32)>

x A Tensor. Must be one of the following types: bfloat16, float16, float32, float64.
lower float, lower bound for setting values to zeros.
upper float, upper bound for setting values to zeros.

A Tensor. Has the same type as x.