fix range op (#37486) (#37611)

a5cf2e30 · Jiawei Wang · GitHub · 3a0c550f · a5cf2e30
隐藏空白更改
内联并排

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(