From 8394699dbbf7b32511a72dd5f76bcdf658ab0011 Mon Sep 17 00:00:00 2001 From: Bai Yifan Date: Fri, 30 Aug 2019 10:03:53 +0800 Subject: [PATCH] add stop_gradient in range_api, test=develop (#19484) --- python/paddle/fluid/layers/tensor.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/paddle/fluid/layers/tensor.py b/python/paddle/fluid/layers/tensor.py index 511f000d91..b0838227f0 100644 --- 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 -- GitLab