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

6 years ago · 8394699dbb
parent e340df013e
commit 8394699dbb
1 changed files with 1 additions and 0 deletions
--- 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