Browse Source

!11831 [Docs] Correct formula in the description of operator Elu

From: @david-he91
Reviewed-by: @ljl0711,@liangchenghui
Signed-off-by: @liangchenghui
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
b036a836a6
1 changed files with 5 additions and 4 deletions
  1. +5
    -4
      mindspore/ops/operations/nn_ops.py

+ 5
- 4
mindspore/ops/operations/nn_ops.py View File

@@ -538,10 +538,11 @@ class Elu(PrimitiveWithInfer):


.. math:: .. math::


\text{x} = \begin{cases}
\alpha * (\exp(\text{x}) - 1), & \text{if x} < \text{0;}\\
\text{x}, & \text{if x} >= \text{0.}
\end{cases}
\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.


The data type of input tensor must be float. The data type of input tensor must be float.




Loading…
Cancel
Save