Browse Source

fix-bug-of-moments-precision-error-in-pynative

tags/v1.1.0
lvliang 5 years ago
parent
commit
4515e6d92d
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