fixed doc for ApplyRMSProp

5 years ago · 2e471f8505
parent a12c8a7868
commit 2e471f8505
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ops/operations/nn_ops.py
+++ b/mindspore/ops/operations/nn_ops.py
@ -1705,8 +1705,8 @@ class ApplyRMSProp(PrimitiveWithInfer):
        - **var** (Tensor) - Weights to be update.
        - **mean_square** (Tensor) - Mean square gradients, must have the same type as `var`.
        - **moment** (Tensor) - Delta of `var`, must have the same type as `var`.
-        - **grad** (Tensor) - Gradients, must have the same type as `var`.
        - **learning_rate** (Union[Number, Tensor]) - Learning rate.
+        - **grad** (Tensor) - Gradients, must have the same type as `var`.
        - **decay** (float) - Decay rate.
        - **momentum** (float) - Momentum.
        - **epsilon** (float) - Ridge term.