tfa.activations.softshrink

View source on GitHub

Soft shrink function.

Computes soft shrink function: x - lower if x < lower, x - upper if x > upper else 0.

x A Tensor. Must be one of the following types: 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.