Browse Source

!9149 Fix bug of moments precision error in pynative

From: @joylvliang
Reviewed-by: @chujinjin,@kisnwang,@jjfeing,@chujinjin
Signed-off-by: @chujinjin
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
d67c54b0dc
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/_extends/builtin_operations.py

+ 2
- 0
mindspore/_extends/builtin_operations.py View File

@@ -157,6 +157,8 @@ def tuple_to_array(x):

def stop_gradient(x):
"""Implement `stop_gradient`."""
if isinstance(x, Tensor):
return Tensor(x.asnumpy())
return x




Loading…
Cancel
Save