From cc6258d6ac15519ba7de59bc7ee154ac0c668a6b Mon Sep 17 00:00:00 2001 From: zhaoting Date: Tue, 14 Apr 2020 09:43:52 +0800 Subject: [PATCH] fix adam notes --- mindspore/ops/operations/nn_ops.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index 538d7f3826..cc6f7da7b0 100644 --- 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