23 lines
666 B
ReStructuredText
23 lines
666 B
ReStructuredText
mindspore.ops.InsertGradientOf
|
||
==============================
|
||
|
||
.. py:class:: mindspore.ops.InsertGradientOf(f)
|
||
|
||
为图节点附加回调函数，将在梯度计算时被调用。
|
||
|
||
.. warning::
|
||
在回调函数中，谨慎使用副作用算子（如TensorDump算子），副作用功能支持尚不完善。
|
||
|
||
参数：
|
||
- **f** (Function) - MindSpore Function 回调函数。
|
||
|
||
输入：
|
||
- **input_x** (Any) - 需要附加回调函数的图节点。
|
||
|
||
输出：
|
||
Tensor，直接返回输入 `input_x` 。该算子不影响前向计算的结果。
|
||
|
||
异常：
|
||
- **TypeError** - `f` 不是MindSpore Function。
|
||
|