xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update apply_proximal_gradient_descent docs

This commit is contained in:
yiyanzhi_akane 2022-07-31 01:08:44 +08:00
parent 649ae3f83c
commit d66c29b13e
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/api/api_python/ops/mindspore.ops.ApplyProximalGradientDescent.rst
View File

@ -22,7 +22,7 @@ mindspore.ops.ApplyProximalGradientDescent
- **alpha** (Union[Number, Tensor]) - 比例系数，必须为标量。数据类型为float16或float32。
- **l1** (Union[Number, Tensor]) - l1正则化强度，必须为标量。数据类型为float16或float32。
- **l2** (Union[Number, Tensor]) - l2正则化强度，必须为标量。数据类型为float16或float32。
- **delta** (Tensor) - 梯度Tensor，shape和数据类型与 `var` 相同。
- **delta** (Tensor) - 梯度Tensor。
输出：
Tensor，更新后的 `var` 。

2
mindspore/python/mindspore/ops/operations/nn_ops.py
View File

@ -6655,7 +6655,7 @@ class ApplyProximalGradientDescent(Primitive):
With float32 or float16 data type.
- **l2** (Union[Number, Tensor]) - l2 regularization strength, must be a scalar.
With float32 or float16 data type.
- **delta** (Tensor) - A tensor for the change, has the same shape and data type as `var`.
- **delta** (Tensor) - A tensor for the change.
Outputs:
Tensor, represents the updated `var`.