fix adam notes

cc6258d6 · zhaoting · fa03a664 · cc6258d6
隐藏空白更改
内联并排

Showing with 5 addition and 1 deletion

mindspore/ops/operations/nn_ops.py mindspore/ops/operations/nn_ops.py +5 -1

未找到文件。
--- 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