forked from mindspore-Ecosystem/mindspore
50 lines
1.4 KiB
ReStructuredText
50 lines
1.4 KiB
ReStructuredText
|
mindspore.ops.Elu
|
|||
|
|
=================
|
|||
|
|
|
|||
|
|
.. py:class:: mindspore.ops.Elu(*args, **kwargs)
|
|||
|
|
|
|||
|
|
指数线性单元激活函数（Exponential Linear Uint activation function）。
|
|||
|
|
|
|||
|
|
对输入的每个元素计算Elu。该激活函数定义如下：
|
|||
|
|
|
|||
|
|
.. 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>`_ 。
|
|||
|
|
|
|||
|
|
**参数：**
|
|||
|
|
|
|||
|
|
**alpha** (float)：Elu的alpha值，数据类型为浮点数。默认值：1.0。
|
|||
|
|
|
|||
|
|
**输入：**
|
|||
|
|
|
|||
|
|
**x** (Tensor) - 用于计算Elu的Tensor，数据类型为float16或float32。shape为 :math:`(N,*)` ，:math:`*` 表示任意的附加维度数。
|
|||
|
|
|
|||
|
|
**输出：**
|
|||
|
|
|
|||
|
|
Tensor，shape和数据类型与 `x` 相同。
|
|||
|
|
|
|||
|
|
**异常：**
|
|||
|
|
|
|||
|
|
- **TypeError** - `alpha` 不是float。
|
|||
|
|
- **TypeError** - `x` 的数据类型既不是float16也不是float32。
|
|||
|
|
- **ValueError** - `alpha` 不等于1.0。
|
|||
|
|
|
|||
|
|
**支持平台：**
|
|||
|
|
|
|||
|
|
``Ascend`` ``GPU`` ``CPU``
|
|||
|
|
|
|||
|
|
**样例：**
|
|||
|
|
|
|||
|
|
>>> x = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32)
|
|||
|
|
>>> elu = ops.Elu()
|
|||
|
|
>>> output = elu(x)
|
|||
|
|
>>> print(output)
|
|||
|
|
[[-0.63212055 4. -0.99966455]
|
|||
|
|
[ 2. -0.99326205 9. ]]
|
|||
|
|
|