forked from mindspore-Ecosystem/mindspore
add nn_ops_api name
This commit is contained in:
parent
cfa8af9668
commit
51c7bd1b29
|
@ -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
|
||||
--------
|
||||
|
||||
|
|
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue