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