fix range op (#37486)

d5c51e62 · Jiawei Wang · GitHub · e76b601b · d5c51e62
显示空白变更内容
内联并排

Showing with 3 addition and 1 deletion

python/paddle/fluid/layers/tensor.py python/paddle/fluid/layers/tensor.py +3 -1

未找到文件。
--- 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(