diff --git a/mindspore/_extends/builtin_operations.py b/mindspore/_extends/builtin_operations.py index a38135c295..494c4e2c38 100644 --- a/mindspore/_extends/builtin_operations.py +++ b/mindspore/_extends/builtin_operations.py @@ -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