mindspore.ops.Elu
- class mindspore.ops.Elu(alpha=1.0)[源代码]
指数线性单元激活函数(Exponential Linear Unit activation function)。
对输入的每个元素计算ELU。该激活函数定义如下:
\[\begin{split}\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.\end{split}\]ELU相关图参见 ELU 。
- 参数:
alpha (float) - Elu的alpha值,数据类型为浮点数。目前只支持alpha等于1.0,默认值:1.0。
- 输入:
input_x (Tensor) - 用于计算Elu的任意维度的Tensor,数据类型为float16,float32或float64。
- 输出:
Tensor,shape和数据类型与 x 相同。
- 异常:
TypeError - alpha 不是float。
TypeError - x 的数据类型既不是float16,float32也不是float64。
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. ]]