mindspore.ops.Elu

class mindspore.ops.Elu(alpha=1.0)

指数线性单元激活函数（Exponential Linear Unit activation function）。

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

$$\begin{array}{r}\text{ELU}(x)=\{\begin{array}{ll}\alpha (e^x-1)& \text{if }x\le 0\\ x& \text{if }x>0\end{array}\end{array}$$

ELU相关图参见 ELU 。

参数：

alpha (float)：Elu的alpha值，数据类型为浮点数。目前只支持alpha等于1.0，默认值：1.0。

输入：

input_x (Tensor) - 用于计算Elu的任意维度的Tensor，数据类型为float16或float32。

输出：

Tensor，shape和数据类型与 x 相同。

异常：

• TypeError - alpha 不是float。

• TypeError - x 的数据类型既不是float16也不是float32。

• ValueError - alpha 不等于1.0。

支持平台：

Ascend GPU CPU

样例：

>>> input_x = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32)
>>> elu = ops.Elu()
>>> output = elu(input_x)
>>> print(output)
[[-0.63212055  4.         -0.99966455]
 [ 2.         -0.99326205  9.        ]]