38 lines
1.2 KiB
ReStructuredText
38 lines
1.2 KiB
ReStructuredText
mindspore.mint.nn.functional.elu
|
||
================================
|
||
|
||
.. py:function:: mindspore.mint.nn.functional.elu(input, alpha=1.0, inplace=False)
|
||
|
||
指数线性单元激活函数。
|
||
|
||
对输入的每个元素计算ELU。该激活函数定义如下：
|
||
|
||
.. math::
|
||
ELU_{i} =
|
||
\begin{cases}
|
||
x_i, &\text{if } x_i \geq 0; \cr
|
||
\alpha * (\exp(x_i) - 1), &\text{otherwise.}
|
||
\end{cases}
|
||
|
||
其中，:math:`x_i` 表示输入的元素，:math:`\alpha` 表示 `alpha` 参数， `alpha` 决定ELU的平滑度。
|
||
|
||
ELU函数图：
|
||
|
||
.. image:: ../images/ELU.png
|
||
:align: center
|
||
|
||
.. warning::
|
||
这是一个实验性API，后续可能修改或删除。
|
||
|
||
参数：
|
||
- **input** (Tensor) - ELU的输入，为任意维度的Tensor。
|
||
- **alpha** (float, 可选) - ELU的alpha值，数据类型为float。默认值： ``1.0``。
|
||
- **inplace** (bool, 可选) - 是否使用原地更新模式，数据类型为bool。默认值： ``False``。
|
||
|
||
返回：
|
||
Tensor，输出的shape和数据类型与 `input` 相同。
|
||
|
||
异常：
|
||
- **RuntimeError** - 如果 `input` 的数据类型不是float16、float32或bfloat16。
|
||
- **TypeError** - 如果 `alpha` 不是float。
|