mindspore.ops.elu

mindspore.ops.elu(input_x, alpha=1.0)[源代码]

指数线性单元激活函数。

对输入的每个元素计算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}\]

其中,\(x\) 表示输入Tensor input_x\(\alpha\) 表示 alpha 参数, alpha 决定ELU的平滑度。 ELU相关图参见 ELU

参数:
  • input_x (Tensor) - ELU的输入,为任意维度的Tensor,数据类型为float16或float32。

  • alpha (float) - ELU的alpha值,数据类型为float,目前仅支持1.0。默认值:1.0。

返回:

Tensor,输出的shape、数据类型与 input_x 相同。

异常:
  • TypeError - 如果 alpha 不是float。

  • TypeError - 如果 input_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)
>>> output = ops.elu(x)
>>> print(output)
[[-0.63212055  4.         -0.99966455]
 [ 2.         -0.99326205  9.        ]]