!27367 add nn_ops_api names

Merge pull request !27367 from xumengjuan1/code_docs_xmj5
This commit is contained in:
i-robot 2021-12-07 11:21:32 +00:00 committed by Gitee
commit 15709e196c
2 changed files with 36 additions and 0 deletions

View File

@ -26,9 +26,28 @@ Non-linear Activations
mindspore.nn.FastGelu
mindspore.nn.HSwish
mindspore.nn.LeakyReLU
mindspore.nn.ReLU
mindspore.nn.ELU
mindspore.nn.GELU
mindspore.nn.Softmax
Utilities
---------
.. cnmsplatformautosummary::
:toctree: nn
mindspore.nn.Flatten
mindspore.nn.Tril
Loss Functions
--------------
.. cnmsplatformautosummary::
:toctree: nn
mindspore.nn.SmoothL1Loss
Optimizer Functions
-------------------
@ -66,6 +85,14 @@ Wrapper Functions
mindspore.nn.WithEvalCell
mindspore.nn.WithLossCell
Math Functions
-----------------
.. cnmsplatformautosummary::
:toctree: nn
mindspore.nn.Moments
Metrics
--------

View File

@ -10,8 +10,11 @@ Neural Network Operators
.. cnmsplatformautosummary::
:toctree: ops
mindspore.ops.BiasAdd
mindspore.ops.Elu
mindspore.ops.GeLU
mindspore.ops.L2Loss
mindspore.ops.L2Normalize
mindspore.ops.PReLU
mindspore.ops.ReLUV2
mindspore.ops.SeLU
@ -28,6 +31,8 @@ Math Operators
mindspore.ops.Div
mindspore.ops.Eps
mindspore.ops.Erf
mindspore.ops.Greater
mindspore.ops.Inv
mindspore.ops.LessEqual
mindspore.ops.Log
mindspore.ops.MatMul
@ -41,9 +46,13 @@ Array Operators
.. cnmsplatformautosummary::
:toctree: ops
mindspore.ops.Eye
mindspore.ops.Fill
mindspore.ops.Gather
mindspore.ops.OnesLike
mindspore.ops.Reshape
mindspore.ops.Size
mindspore.ops.Squeeze
mindspore.ops.Tile
mindspore.ops.ZerosLike