Browse Source

fix adam notes

tags/v0.2.0-alpha
zhaoting 5 years ago
parent
commit
cc6258d6ac
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      mindspore/ops/operations/nn_ops.py

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

@@ -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


Loading…
Cancel
Save