fixed doc for CumSum

5 years ago · 019911c56a
parent 2e471f8505
commit 019911c56a
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ops/operations/math_ops.py
+++ b/mindspore/ops/operations/math_ops.py
@ -636,7 +636,7 @@ class CumSum(PrimitiveWithInfer):

    Inputs:
        - **input** (Tensor) - The input tensor to accumulate.
-        - **axis**  (int) - The axis to accumulate the tensor's value.
+        - **axis**  (int) - The axis to accumulate the tensor's value. Only constant value is allowed.

    Outputs:
        Tensor, the shape of the output tensor is consistent with the input tensor's.