fix range op (#37486) (#37611)

PaddlePaddle · Nov 30, 2021 · a5cf2e3 · a5cf2e3
1 parent 3a0c550
commit a5cf2e3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/paddle/fluid/layers/tensor.py b/python/paddle/fluid/layers/tensor.py
@@ -1433,7 +1433,9 @@ def range(start, end, step, dtype, name=None):
         step = cast(step, dtype)
 
     if in_dygraph_mode():
-        return _C_ops.range(start, end, step)
+        out = _C_ops.range(start, end, step)
+        out.stop_gradient = True
+        return out
 
     out_shape = None
     if not isinstance(start, Variable) and not isinstance(