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

38 lines
1.1 KiB
ReStructuredText
Raw Normal View History

2021-12-04 13:55:42 +08:00
mindspore.nn.ELU
=================
.. py:class:: mindspore.nn.ELU(alpha=1.0)
指数线性单元激活函数Exponential Linear Unit activation function
2021-12-04 13:55:42 +08:00
对输入的每个元素计算ELU。该激活函数定义如下
2021-12-04 20:36:47 +08:00
2021-12-04 13:55:42 +08:00
.. math::
E_{i} =
\begin{cases}
x_i, &\text{if } x_i \geq 0; \cr
\alpha * (\exp(x_i) - 1), &\text{otherwise.}
2021-12-04 13:55:42 +08:00
\end{cases}
其中,:math:`x_i` 表示输入的元素,:math:`\alpha` 表示 `alpha` 参数。
2021-12-04 13:55:42 +08:00
ELU相关图参见 `ELU <https://en.wikipedia.org/wiki/Activation_function#/media/File:Activation_elu.svg>`_
**参数:**
- **alpha** (`float`) ELU的alpha值数据类型为浮点数。默认值1.0。
**输入:**
- **x** Tensor - 用于计算ELU的任意维度的Tensor数据类型为float16或float32。
2021-12-04 13:55:42 +08:00
**输出:**
Tensor数据类型和shape与 `x` 相同。
2021-12-04 13:55:42 +08:00
**异常:**
- **TypeError** - `alpha` 不是浮点数。
- **TypeError** - `x` 的数据类型既不是float16也不是float32。
- **ValueError** - `alpha` 不等于1.0。