fix adam notes

5 years ago · cc6258d6ac
parent fa03a66433
commit cc6258d6ac
1 changed files with 5 additions and 1 deletions
--- a/mindspore/ops/operations/nn_ops.py
+++ b/mindspore/ops/operations/nn_ops.py
@ -2136,7 +2136,11 @@ class Adam(PrimitiveWithInfer):
        - **gradient** (Tensor) - Gradients.

    Outputs:
-        Tensor, has the same shape and data type as `var`.
+        Tuple of 3 Tensor, the updated parameters.
+
+        - **var** (Tensor) - The same shape and data type as `var`.
+        - **m** (Tensor) - The same shape and data type as `m`.
+        - **v** (Tensor) - The same shape and data type as `v`.
    """

    @prim_attr_register