diff --git a/mindspore/nn/grad/cell_grad.py b/mindspore/nn/grad/cell_grad.py index d7d13440a19..1c5f1316751 100644 --- a/mindspore/nn/grad/cell_grad.py +++ b/mindspore/nn/grad/cell_grad.py @@ -91,7 +91,7 @@ class Jvp(Cell): >>> v = Tensor(np.array([[1, 1], [1, 1]]).astype(np.float32)) >>> output = Jvp(Net())(x, y, (v, v)) >>> print(output[0]) - [[2, 10], [20, 68]] + [[2, 10], [30, 68]] >>> print(output[1]) [[4, 13], [28, 49]] """ @@ -203,7 +203,7 @@ class Vjp(Cell): >>> v = Tensor(np.array([[1, 1], [1, 1]]).astype(np.float32)) >>> output = Vjp(Net())(x, y, v) >>> print(output[0]) - [[2, 10], [20, 68]] + [[2, 10], [30, 68]] >>> print(output[1]) ([[3, 12], [27, 48]], [[1, 1], [1, 1]]) """