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

!9149 Fix bug of moments precision error in pynative 

From: @joylvliang
Reviewed-by: @chujinjin,@kisnwang,@jjfeing,@chujinjin
Signed-off-by: @chujinjin
This commit is contained in:
mindspore-ci-bot 2020-11-30 09:08:47 +08:00 committed by Gitee
parent 9c38a63f18 4515e6d92d
commit d67c54b0dc
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/_extends/builtin_operations.py
View File

@ -157,6 +157,8 @@ def tuple_to_array(x):
def stop_gradient(x): def stop_gradient(x):
"""Implement `stop_gradient`.""" """Implement `stop_gradient`."""
if isinstance(x, Tensor):
return Tensor(x.asnumpy())
return x return x