Browse Source

fix docstring

fix docstring
tags/v1.1.0
Peilin Wang 5 years ago
parent
commit
c81a4af73a
1 changed files with 6 additions and 3 deletions
  1. +6
    -3
      mindspore/ops/operations/nn_ops.py

+ 6
- 3
mindspore/ops/operations/nn_ops.py View File

@@ -5171,7 +5171,9 @@ class SparseApplyProximalAdagrad(PrimitiveWithCheck):
- **l2** (Union[Number, Tensor]) - l2 regularization strength, must be a float number or
a scalar tensor with float16 or float32 data type..
- **grad** (Tensor) - A tensor of the same type as `var`, for the gradient.
- **indices** (Tensor) - A vector of indices in the first dimension of `var` and `accum`.
- **indices** (Tensor) - A tensor of indices in the first dimension of `var` and `accum`.
If there are duplicates in `indices`, the behavior is undefined. Must be one of the
following types: int16, int32, int64, uint16, uint32, uint64.

Outputs:
Tuple of 2 tensors, the updated parameters.
@@ -5840,8 +5842,9 @@ class SparseApplyFtrl(PrimitiveWithCheck):
- **accum** (Parameter) - The accumulation to be updated, must be same data type and shape as `var`.
- **linear** (Parameter) - the linear coefficient to be updated, must be the same data type and shape as `var`.
- **grad** (Tensor) - A tensor of the same type as `var`, for the gradient.
- **indices** (Tensor) - A vector of indices in the first dimension of `var` and `accum`.
The shape of `indices` must be the same as `grad` in the first dimension. The type must be int32.
- **indices** (Tensor) - A tensor of indices in the first dimension of `var` and `accum`.
The shape of `indices` must be the same as `grad` in the first dimension. If there are
duplicates in `indices`, the behavior is undefined. The type must be int32 or int64.

Outputs:
- **var** (Tensor) - Tensor, has the same shape and data type as `var`.


Loading…
Cancel
Save