mindspore/docs/api/api_python/nn/mindspore.nn.Tanhshrink.rst

26 lines
747 B
ReStructuredText
Raw Normal View History

mindspore.nn.Tanhshrink
=======================
.. py:class:: mindspore.nn.Tanhshrink
Tanhshrink激活函数。
按元素计算Tanhshrink函数返回一个新的Tensor。
Tanhshrink函数定义为
.. math::
tanhshrink(x_i) =x_i- \frac{\exp(x_i) - \exp(-x_i)}{\exp(x_i) + \exp(-x_i)} = x_i-\frac{\exp(2x_i) - 1}{\exp(2x_i) + 1},
其中 :math:`x_i` 是输入Tensor的元素。
输入:
- **x** (Tensor) - 任意维度的Tensor数据类型为float16或float32的输入。
2022-06-27 16:24:39 +08:00
输出:
Tensor数据类型和shape与 `x` 的相同。
异常:
- **TypeError** - `x` 不是一个Tensor。
- **TypeError** - `x` 的数据类型既不是float16也不是float32。