p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Docs] Correct formula in the description of operator Elu

This commit is contained in:
hedongdong 2021-01-29 16:07:28 +08:00
parent 424e68a803
commit de2db9d5a6
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/ops/operations/nn_ops.py
View File

@ -538,10 +538,11 @@ class Elu(PrimitiveWithInfer):
.. math:: .. math::
\text{x} = \begin{cases} \text{ELU}(x)= \left\{
\alpha * (\exp(\text{x}) - 1), & \text{if x} < \text{0;}\\ \begin{array}{align}
\text{x}, & \text{if x} >= \text{0.} \alpha(e^{x} - 1) & \text{if } x \le 0\\
\end{cases} 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.