mindspore/docs/api/api_python/ops/mindspore.ops.Elu.rst

33 lines
1.1 KiB
ReStructuredText
Raw Normal View History

2021-12-07 12:17:56 +08:00
mindspore.ops.Elu
=================
2021-12-21 18:09:08 +08:00
.. py:class:: mindspore.ops.Elu(alpha=1.0)
2021-12-07 12:17:56 +08:00
指数线性单元激活函数Exponential Linear Unit activation function
2021-12-07 12:17:56 +08:00
2021-12-21 18:09:08 +08:00
对输入的每个元素计算ELU。该激活函数定义如下
2021-12-07 12:17:56 +08:00
.. math::
\text{ELU}(x)= \left\{
\begin{array}{align}
\alpha(e^{x} - 1) & \text{if } x \le 0\\
x & \text{if } x \gt 0\\
\end{array}\right.
ELU相关图参见 `ELU <https://en.wikipedia.org/wiki/Activation_function#/media/File:Activation_elu.svg>`_
2022-07-25 15:09:45 +08:00
参数:
- **alpha** (float) - Elu的alpha值数据类型为浮点数。目前只支持alpha等于1.0默认值1.0。
2021-12-07 12:17:56 +08:00
2022-07-25 15:09:45 +08:00
输入:
- **input_x** (Tensor) - 用于计算Elu的任意维度的Tensor数据类型为float16或float32。
2021-12-07 12:17:56 +08:00
2022-07-25 15:09:45 +08:00
输出:
Tensorshape和数据类型与 `x` 相同。
2021-12-07 12:17:56 +08:00
2022-07-25 15:09:45 +08:00
异常:
- **TypeError** - `alpha` 不是float。
- **TypeError** - `x` 的数据类型既不是float16也不是float32。
- **ValueError** - `alpha` 不等于1.0。