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.

mindspore.nn.rst 3.3 kB

4 years ago
4 years ago
4 years ago
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176
  1. mindspore.nn
  2. =============
  3. Cell
  4. ----
  5. .. cnmsplatformautosummary::
  6. :toctree: nn
  7. mindspore.nn.Cell
  8. 容器
  9. -----------
  10. .. cnmsplatformautosummary::
  11. :toctree: nn
  12. mindspore.nn.CellList
  13. mindspore.nn.SequentialCell
  14. 卷积层
  15. --------------------
  16. .. cnmsplatformautosummary::
  17. :toctree: nn
  18. mindspore.nn.Conv1d
  19. mindspore.nn.Conv1dTranspose
  20. mindspore.nn.Conv2d
  21. mindspore.nn.Conv2dTranspose
  22. mindspore.nn.Conv3d
  23. mindspore.nn.Conv3dTranspose
  24. Gradient
  25. ---------
  26. .. cnmsplatformautosummary::
  27. :toctree: nn
  28. mindspore.nn.Jvp
  29. mindspore.nn.Vjp
  30. 非线性激活函数
  31. ----------------------
  32. .. cnmsplatformautosummary::
  33. :toctree: nn
  34. mindspore.nn.FastGelu
  35. mindspore.nn.HShrink
  36. mindspore.nn.HSigmoid
  37. mindspore.nn.HSwish
  38. mindspore.nn.LeakyReLU
  39. mindspore.nn.LogSigmoid
  40. mindspore.nn.LogSoftmax
  41. mindspore.nn.ReLU
  42. mindspore.nn.ELU
  43. mindspore.nn.GELU
  44. mindspore.nn.Sigmoid
  45. mindspore.nn.Softmax
  46. mindspore.nn.Tanh
  47. Utilities
  48. ---------
  49. .. cnmsplatformautosummary::
  50. :toctree: nn
  51. mindspore.nn.Flatten
  52. mindspore.nn.Tril
  53. 损失函数
  54. --------------
  55. .. cnmsplatformautosummary::
  56. :toctree: nn
  57. mindspore.nn.L1Loss
  58. mindspore.nn.MSELoss
  59. mindspore.nn.SmoothL1Loss
  60. Optimizer Functions
  61. -------------------
  62. .. cnmsplatformautosummary::
  63. :toctree: nn
  64. mindspore.nn.Optimizer
  65. mindspore.nn.Adagrad
  66. mindspore.nn.Adam
  67. mindspore.nn.AdamOffload
  68. mindspore.nn.AdamWeightDecay
  69. mindspore.nn.FTRL
  70. mindspore.nn.LARS
  71. mindspore.nn.Lamb
  72. mindspore.nn.LazyAdam
  73. mindspore.nn.Momentum
  74. mindspore.nn.ProximalAdagrad
  75. mindspore.nn.RMSProp
  76. mindspore.nn.SGD
  77. Wrapper Functions
  78. -----------------
  79. .. cnmsplatformautosummary::
  80. :toctree: nn
  81. mindspore.nn.DistributedGradReducer
  82. mindspore.nn.DynamicLossScaleUpdateCell
  83. mindspore.nn.FixedLossScaleUpdateCell
  84. mindspore.nn.ForwardValueAndGrad
  85. mindspore.nn.PipelineCell
  86. mindspore.nn.TrainOneStepCell
  87. mindspore.nn.TrainOneStepWithLossScaleCell
  88. mindspore.nn.WithEvalCell
  89. mindspore.nn.WithLossCell
  90. Math Functions
  91. -----------------
  92. .. cnmsplatformautosummary::
  93. :toctree: nn
  94. mindspore.nn.Moments
  95. Metrics
  96. --------
  97. .. cnmsautosummary::
  98. :toctree: nn
  99. mindspore.nn.Accuracy
  100. mindspore.nn.F1
  101. mindspore.nn.Fbeta
  102. mindspore.nn.Loss
  103. mindspore.nn.MAE
  104. mindspore.nn.MSE
  105. mindspore.nn.Metric
  106. mindspore.nn.Precision
  107. mindspore.nn.Recall
  108. mindspore.nn.Top1CategoricalAccuracy
  109. mindspore.nn.Top5CategoricalAccuracy
  110. mindspore.nn.TopKCategoricalAccuracy
  111. mindspore.nn.get_metric_fn
  112. mindspore.nn.names
  113. mindspore.nn.rearrange_inputs
  114. Dynamic Learning Rate
  115. ---------------------
  116. LearningRateSchedule
  117. ^^^^^^^^^^^^^^^^^^^^^
  118. .. cnmsplatformautosummary::
  119. :toctree: nn
  120. mindspore.nn.CosineDecayLR
  121. mindspore.nn.ExponentialDecayLR
  122. mindspore.nn.InverseDecayLR
  123. mindspore.nn.NaturalExpDecayLR
  124. mindspore.nn.PolynomialDecayLR
  125. mindspore.nn.WarmUpLR
  126. Dynamic LR
  127. ^^^^^^^^^^
  128. .. cnmsautosummary::
  129. :toctree: nn
  130. mindspore.nn.cosine_decay_lr
  131. mindspore.nn.exponential_decay_lr
  132. mindspore.nn.inverse_decay_lr
  133. mindspore.nn.natural_exp_decay_lr
  134. mindspore.nn.piecewise_constant_lr
  135. mindspore.nn.polynomial_decay_lr
  136. mindspore.nn.warmup_lr