p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!27045 Fix wrong return value of nn.Jvp and nn.Vjp examples. 

Merge pull request !27045 from LiangZhibo/code_docs_grad
This commit is contained in:
i-robot 2021-12-02 01:01:41 +00:00 committed by Gitee
parent c0d4e5cf0f 9339389e94
commit 0a05868d07
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)) >>> v = Tensor(np.array([[1, 1], [1, 1]]).astype(np.float32))
>>> output = Jvp(Net())(x, y, (v, v)) >>> output = Jvp(Net())(x, y, (v, v))
>>> print(output[0]) >>> print(output[0])
[[2, 10], [20, 68]] [[2, 10], [30, 68]]
>>> print(output[1]) >>> print(output[1])
[[4, 13], [28, 49]] [[4, 13], [28, 49]]
""" """
@ -203,7 +203,7 @@ class Vjp(Cell):
>>> v = Tensor(np.array([[1, 1], [1, 1]]).astype(np.float32)) >>> v = Tensor(np.array([[1, 1], [1, 1]]).astype(np.float32))
>>> output = Vjp(Net())(x, y, v) >>> output = Vjp(Net())(x, y, v)
>>> print(output[0]) >>> print(output[0])
[[2, 10], [20, 68]] [[2, 10], [30, 68]]
>>> print(output[1]) >>> print(output[1])
([[3, 12], [27, 48]], [[1, 1], [1, 1]]) ([[3, 12], [27, 48]], [[1, 1], [1, 1]])
""" """