tfa.activations.softshrink

View source on GitHub

Soft shrink function.

tfa.activations.softshrink(
    x,
    lower=-0.5,
    upper=0.5
)

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

Args:

  • 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.

Returns:

A Tensor. Has the same type as x.