add stop_gradient in range_api, test=develop (#19484)

8394699d · Bai Yifan · whs · e340df01 · 8394699d
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

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

未找到文件。
--- a/python/paddle/fluid/layers/tensor.py
+++ b/python/paddle/fluid/layers/tensor.py
@@ -874,6 +874,7 @@ def range(start, end, step, dtype):
                'End': end,
                'Step': step},
        outputs={'Out': [out]})
+    out.stop_gradient = True
    return out