diff --git a/tutorials/source_zh_cn/use/custom_operator.md b/tutorials/source_zh_cn/use/custom_operator.md index be67329e64d70c880da21abc092a139a7dcd3323..18004d1f8817d6f9a7f32095065aa4da4e16ecc8 100644 --- a/tutorials/source_zh_cn/use/custom_operator.md +++ b/tutorials/source_zh_cn/use/custom_operator.md @@ -223,14 +223,15 @@ class CusSquare(PrimitiveWithInfer): return bprop ``` -在`test_square.py`文件中定义反向用例。其中,`grad_with_sens`是`GradOperation`类的实例化对象。 +在`test_square.py`文件中定义反向用例。 ```python from mindspore.ops import composite as C def test_grad_net(): x = np.array([1.0, 4.0, 9.0]).astype(np.float32) sens = np.array([1.0, 1.0, 1.0]).astype(np.float32) square = Net() - dx = C.grad_with_sens(square)(Tensor(x), Tensor(sens)) + grad = C.GradOperation('grad_with_sens', sens_param=True) + dx = grad(square)(Tensor(x), Tensor(sens)) print("x: ", x) print("dx: ", dx) ```