Modify the comment of InsertGradientOf

5 years ago · c24e90a3a6
parent 3c1b8308cf
commit c24e90a3a6
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ops/operations/debug_ops.py
+++ b/mindspore/ops/operations/debug_ops.py
@ -191,7 +191,7 @@ class InsertGradientOf(PrimitiveWithInfer):
        f (Function): MindSpore's Function. Callback function.

    Inputs:
-        - **input_x** (Tensor) - The graph node to attach to.
+        - **input_x** (Any) - The graph node to attach to.

    Outputs:
        Tensor, returns `input_x` directly. `InsertGradientOf` does not affect the forward result.