zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 18fa59e090
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '18fa59e090'
${ noResults }
mindspore2022/docs/api/api_python/mindspore.nn.rst

178 lines
3.4 kB
Raw Blame History 

  1. mindspore.nn

  2. =============

  3. 

  4. Cell

  5. ----

  6. 

  7. .. cnmsplatformautosummary::

  8.     :toctree: nn

  9. 

  10.     mindspore.nn.Cell

  11. 

  12. Containers

  13. -----------

  14. 

  15. .. cnmsplatformautosummary::

  16.     :toctree: nn

  17. 

  18.     mindspore.nn.CellList

  19.     mindspore.nn.SequentialCell

  20. 

  21. Convolution Layers

  22. --------------------

  23. 

  24. .. cnmsplatformautosummary::

  25.     :toctree: nn

  26. 

  27.     mindspore.nn.Conv1d

  28.     mindspore.nn.Conv1dTranspose

  29.     mindspore.nn.Conv2d

  30.     mindspore.nn.Conv2dTranspose

  31.     mindspore.nn.Conv3d

  32.     mindspore.nn.Conv3dTranspose

  33. 

  34. Gradient

  35. ---------

  36. 

  37. .. cnmsplatformautosummary::

  38.     :toctree: nn

  39. 

  40.     mindspore.nn.Jvp

  41.     mindspore.nn.Vjp

  42. 

  43. Non-linear Activations

  44. ----------------------

  45. 

  46. .. cnmsplatformautosummary::

  47.     :toctree: nn

  48. 

  49.     mindspore.nn.FastGelu

  50.     mindspore.nn.HShrink

  51.     mindspore.nn.HSigmoid

  52.     mindspore.nn.HSwish

  53.     mindspore.nn.LeakyReLU

  54.     mindspore.nn.LogSigmoid

  55.     mindspore.nn.LogSoftmax

  56.     mindspore.nn.ReLU

  57.     mindspore.nn.ELU

  58.     mindspore.nn.GELU

  59.     mindspore.nn.Sigmoid

  60.     mindspore.nn.Softmax

  61.     mindspore.nn.Tanh

  62. 

  63. Utilities

  64. ---------

  65. 

  66. .. cnmsplatformautosummary::

  67.     :toctree: nn

  68. 

  69.     mindspore.nn.Flatten

  70.     mindspore.nn.Tril

  71. 

  72. Loss Functions

  73. --------------

  74. 

  75. .. cnmsplatformautosummary::

  76.     :toctree: nn

  77. 

  78.     mindspore.nn.L1Loss

  79.     mindspore.nn.MAELoss

  80.     mindspore.nn.MSELoss

  81.     mindspore.nn.SmoothL1Loss

  82. 

  83. Optimizer Functions

  84. -------------------

  85. 

  86. .. cnmsplatformautosummary::

  87.     :toctree: nn

  88. 

  89.     mindspore.nn.Optimizer

  90.     mindspore.nn.Adagrad

  91.     mindspore.nn.Adam

  92.     mindspore.nn.AdamOffload

  93.     mindspore.nn.AdamWeightDecay

  94.     mindspore.nn.FTRL

  95.     mindspore.nn.LARS

  96.     mindspore.nn.Lamb

  97.     mindspore.nn.LazyAdam

  98.     mindspore.nn.Momentum

  99.     mindspore.nn.ProximalAdagrad

  100.     mindspore.nn.RMSProp

  101.     mindspore.nn.SGD

  102. 

  103. Wrapper Functions

  104. -----------------

  105. 

  106. .. cnmsplatformautosummary::

  107.     :toctree: nn

  108. 

  109.     mindspore.nn.DistributedGradReducer

  110.     mindspore.nn.DynamicLossScaleUpdateCell

  111.     mindspore.nn.FixedLossScaleUpdateCell

  112.     mindspore.nn.ForwardValueAndGrad

  113.     mindspore.nn.PipelineCell

  114.     mindspore.nn.TrainOneStepCell

  115.     mindspore.nn.TrainOneStepWithLossScaleCell

  116.     mindspore.nn.WithEvalCell

  117.     mindspore.nn.WithLossCell

  118. 

  119. Math Functions

  120. -----------------

  121. 

  122. .. cnmsplatformautosummary::

  123.     :toctree: nn

  124. 

  125.     mindspore.nn.Moments

  126. 

  127. Metrics

  128. --------

  129. 

  130. .. cnmsautosummary::

  131.     :toctree: nn

  132. 

  133.     mindspore.nn.Accuracy

  134.     mindspore.nn.F1

  135.     mindspore.nn.Fbeta

  136.     mindspore.nn.Loss

  137.     mindspore.nn.MAE

  138.     mindspore.nn.MSE

  139.     mindspore.nn.Metric

  140.     mindspore.nn.Precision

  141.     mindspore.nn.Recall

  142.     mindspore.nn.Top1CategoricalAccuracy

  143.     mindspore.nn.Top5CategoricalAccuracy

  144.     mindspore.nn.TopKCategoricalAccuracy

  145.     mindspore.nn.get_metric_fn

  146.     mindspore.nn.names

  147.     mindspore.nn.rearrange_inputs

  148. 

  149. Dynamic Learning Rate

  150. ---------------------

  151. 

  152. LearningRateSchedule

  153. ^^^^^^^^^^^^^^^^^^^^^

  154. 

  155. .. cnmsplatformautosummary::

  156.     :toctree: nn

  157. 

  158.     mindspore.nn.CosineDecayLR

  159.     mindspore.nn.ExponentialDecayLR

  160.     mindspore.nn.InverseDecayLR

  161.     mindspore.nn.NaturalExpDecayLR

  162.     mindspore.nn.PolynomialDecayLR

  163.     mindspore.nn.WarmUpLR

  164. 

  165. Dynamic LR

  166. ^^^^^^^^^^

  167. 

  168. .. cnmsautosummary::

  169.     :toctree: nn

  170. 

  171.     mindspore.nn.cosine_decay_lr

  172.     mindspore.nn.exponential_decay_lr

  173.     mindspore.nn.inverse_decay_lr

  174.     mindspore.nn.natural_exp_decay_lr

  175.     mindspore.nn.piecewise_constant_lr

  176.     mindspore.nn.polynomial_decay_lr

  177.     mindspore.nn.warmup_lr