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

!35182 [bug] fix taylor rule of log 

Merge pull request !35182 from chenzhuo/taylor_log
This commit is contained in:
i-robot 2022-05-31 01:33:29 +00:00 committed by Gitee
parent 5101633394 73038f9d02
commit 8cabd14dcd
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/ops/_grad/taylor_rule.py
View File

@ -165,7 +165,7 @@ def taylor_log(self):
for k in range(1, series_num + 1):
for i in range(1, k):
tmp = i * inputs[k - i] * series[i] / (factorial[k - i] * factorial[i])
series[k] -= tmp
series[k] += tmp
series[k] = (inputs[k] - factorial[k - 1] * series[k]) / inputs[0]
return series

6
tests/st/gradient/test_taylor_differentiation_graph.py
View File

@ -102,10 +102,10 @@ def test_jet_single_input_single_output_with_scalar_graph_mode():
primals = Tensor([2., 2.])
series = Tensor([[1., 1.], [0., 0.], [0., 0.]])
net = SingleInputSingleOutputWithScalarNet()
expected_primals = np.array([10.328085, 10.328085]).astype(np.float32)
expected_series = np.array([[-3.1220534, -3.1220534], [2.943144, 2.943144],
[-4.551988, -4.551988]]).astype(np.float32)
out_primals, out_series = jet(net, primals, series)
expected_primals = np.array([10.328085, 10.328085]).astype(np.float32)
expected_series = np.array([[-3.1220534, -3.1220534], [6.0652323, 6.0652323],
[-18.06463, -18.06463]]).astype(np.float32)
assert np.allclose(out_series.asnumpy(), expected_series, atol=1.e-4)
assert np.allclose(out_primals.asnumpy(), expected_primals, atol=1.e-4)