From b4dddb2994ffe64e43132d44276fd65ca3c57aa1 Mon Sep 17 00:00:00 2001 From: Yang Yu Date: Tue, 7 Nov 2017 19:31:48 -0800 Subject: [PATCH] Fix Unittest --- python/paddle/v2/framework/layers.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/paddle/v2/framework/layers.py b/python/paddle/v2/framework/layers.py index e235ff369e..8fc34501c6 100644 --- a/python/paddle/v2/framework/layers.py +++ b/python/paddle/v2/framework/layers.py @@ -801,12 +801,13 @@ def zeros(shape, dtype, main_program=None): def increment(x, value=1.0, main_program=None): helper = LayerHelper("increment", **locals()) + out = helper.create_tmp_variable(dtype=x.data_type) helper.append_op( type='increment', inputs={'X': [x]}, - outputs={'Out': [x]}, + outputs={'Out': [out]}, attrs={'step': value}) - return x + return out def array_write(x, i, array=None, main_program=None): -- GitLab