tfa.activations.tanhshrink

View source on GitHub

Tanh shrink function.

tfa.activations.tanhshrink(
    x: tfa.types.TensorLike
) -> tf.Tensor

Applies the element-wise function:

\[ \mathrm{tanhshrink}(x) = x - \tanh(x). \]

Usage:

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

Args
x	A Tensor. Must be one of the following types: bfloat16, float16, float32, float64.

Returns
A Tensor. Has the same type as x.