From 5c6745b3f50f12b19f8aa0417b0ae028711d68cb Mon Sep 17 00:00:00 2001 From: lvliang Date: Fri, 26 Mar 2021 09:34:30 +0800 Subject: [PATCH] update_case_for_interface_forwardvalueandgrad --- mindspore/nn/wrap/cell_wrapper.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/mindspore/nn/wrap/cell_wrapper.py b/mindspore/nn/wrap/cell_wrapper.py index b143a736bc..30e760ae09 100644 --- a/mindspore/nn/wrap/cell_wrapper.py +++ b/mindspore/nn/wrap/cell_wrapper.py @@ -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):