mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

Fix wrong code example result for jvp and vjp

This commit is contained in:
l00591931 2021-12-01 14:18:39 +08:00
parent e67f83fdf1
commit 9339389e94
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/nn/grad/cell_grad.py
View File

@ -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]])
"""