add nn_ops_api name

This commit is contained in:
xumengjuan1 2021-12-07 17:45:17 +08:00
parent cfa8af9668
commit 51c7bd1b29
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