a946079208
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!14138 Update case for interface forwardvalueandgrad 

From: @joylvliang
Reviewed-by: @ginfung,@zhunaipan
Signed-off-by: @ginfung
This commit is contained in:
mindspore-ci-bot 2021-03-27 10:08:22 +08:00 committed by Gitee
parent d9d3aaed19 5c6745b3f5
commit 1892a629c8
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/nn/wrap/cell_wrapper.py
View File

@ -238,11 +238,11 @@ class ForwardValueAndGrad(Cell):
>>> labels = Tensor(np.zeros([1, 2]).astype(np.float32))
>>> result = train_network(inputs, labels)
>>> print(result)
(Tensor(shape=[1], dtype=Float32, value=[0]), ((Tensor(shape=[1, 2], dtype=Float32, value=
[[1, 1]]), Tensor(shape=[1, 2], dtype=Float32, value=
[[0, 0]])), (Tensor(shape=[2, 2], dtype=Float32, value=
[[0.5, 0.5],
[0.5, 0.5]]),)))
(Tensor(shape=[1], dtype=Float32, value=[0.00000000e+00]), ((Tensor(shape=[1, 2], dtype=Float32, value=
[[1.00000000e+00, 1.00000000e+00]]), Tensor(shape=[1, 2], dtype=Float32, value=
[[0.00000000e+00, 0.00000000e+00]])), (Tensor(shape=[2, 2], dtype=Float32, value=
[[5.00000000e-01, 5.00000000e-01],
[5.00000000e-01, 5.00000000e-01]]),)))
"""
def __init__(self, network, weights=None, get_all=False, get_by_list=False, sens_param=False):