diff --git a/mindspore/ops/operations/debug_ops.py b/mindspore/ops/operations/debug_ops.py
index c4fbddd38e7f2279e6b87dd27dc476f97246fb12..c6b635a69fb8dc0d87aebc21437b967db342b35e 100644
--- 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.