mindspore.mint.nn.functional.elu_

mindspore.mint.nn.functional.elu_(input, alpha=1.0)

指数线性单元激活函数。

对输入的每个元素原地计算ELU。该激活函数定义如下：

\[ELU_{i} = \begin{cases} x_i, &\text{if } x_i \geq 0; \cr \alpha * (\exp(x_i) - 1), &\text{otherwise.} \end{cases}\]

其中，\(x_i\) 表示输入的元素，\(\alpha\) 表示 alpha 参数， alpha 决定ELU的平滑度。

ELU函数图：

警告

这是一个实验性API，后续可能修改或删除。

参数：

• input (Tensor) - ELU的输入，为任意维度的Tensor。

• alpha (float, 可选) - ELU的alpha值，数据类型为float且必须大于0。默认值： 1.0 。

返回：

Tensor，输出的shape和数据类型与 input 相同。

异常：

• RuntimeError - 如果 input 的数据类型不是float16、float32或bfloat16。

• TypeError - 如果 alpha 的数据类型不是float。

支持平台：

Ascend

样例：

>>> import mindspore
>>> from mindspore import Tensor, mint
>>> import numpy as np
>>> input = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float32)
>>> mint.nn.functional.elu_(input)
>>> print(input)
[-0.63212055  -0.86466473  0.  2.  1.]