2022-06-22 14:52:28 +08:00
|
|
|
|
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的元素。
|
|
|
|
|
|
|
2022-07-25 16:47:23 +08:00
|
|
|
|
输入:
|
|
|
|
|
|
- **x** (Tensor) - 任意维度的Tensor,数据类型为float16或float32的输入。
|
2022-06-27 16:24:39 +08:00
|
|
|
|
|
2022-07-25 16:47:23 +08:00
|
|
|
|
输出:
|
|
|
|
|
|
Tensor,数据类型和shape与 `x` 的相同。
|
2022-06-22 14:52:28 +08:00
|
|
|
|
|
2022-07-25 16:47:23 +08:00
|
|
|
|
异常:
|
|
|
|
|
|
- **TypeError** - `x` 不是一个Tensor。
|
|
|
|
|
|
- **TypeError** - `x` 的数据类型既不是float16也不是float32。
|