From 30adea0a2378f487e66f117d285708e6b5ed271c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=9F=B3=E6=99=93=E4=BC=9F?= <39303645+Shixiaowei02@users.noreply.github.com> Date: Mon, 23 Sep 2019 10:30:11 +0800 Subject: [PATCH] tensor_array_to_tensor_op.cc, test=develop (#19289) --- .../fluid/operators/tensor_array_to_tensor_op.cc | 15 +++------------ 1 file changed, 3 insertions(+), 12 deletions(-) diff --git a/paddle/fluid/operators/tensor_array_to_tensor_op.cc b/paddle/fluid/operators/tensor_array_to_tensor_op.cc index 2b83c42f205..8cba4961153 100644 --- a/paddle/fluid/operators/tensor_array_to_tensor_op.cc +++ b/paddle/fluid/operators/tensor_array_to_tensor_op.cc @@ -105,15 +105,7 @@ class LoDTensorArray2TensorOp : public framework::OperatorBase { auto out_inx_dim = out_inx.dims(); out_inx_dim[0] = inx.size(); out_inx.Resize(out_inx_dim); - - auto &local_scope = scope.NewScope(); - std::string var_name = "out_index"; - framework::Variable *tmp_index_var = local_scope.Var(var_name); - auto &tmp_index_tensor = - *(tmp_index_var->GetMutable()); - tmp_index_tensor.Resize(out_inx_dim); - int *tmp_index_data = - tmp_index_tensor.mutable_data(platform::CPUPlace()); + int *tmp_index_data = out_inx.mutable_data(platform::CPUPlace()); auto out_dims = inx[0].dims(); size_t out_dim_sum = 0; @@ -122,18 +114,17 @@ class LoDTensorArray2TensorOp : public framework::OperatorBase { out_dim_sum += inx_dims[axis]; tmp_index_data[index] = inx_dims[axis]; } - out_inx.ShareDataWith(tmp_index_tensor); // get input array items' dims out_dims[axis] = out_dim_sum; out.Resize(out_dims); - LodTensorArray2LodTensorVector(local_scope, base_name, Input("X"), &names); + LodTensorArray2LodTensorVector(scope, base_name, Input("X"), &names); // Invoke concat Op auto concat_op = framework::OpRegistry::CreateOp( "concat", {{"X", names}}, {{"Out", {Output("Out")}}}, attrs); - concat_op->Run(local_scope, place); + concat_op->Run(scope, place); } }; -- GitLab