Modify the comment of InsertGradientOf

This commit is contained in:
simson 2020-06-19 10:21:49 +08:00
parent 3c1b8308cf
commit c24e90a3a6
1 changed files with 1 additions and 1 deletions

View File

@ -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.