fix bug (#5106)

PaddlePaddle · Oct 26, 2017 · 39a6f43 · 39a6f43
1 parent 6cc2ce0
commit 39a6f43
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/paddle/framework/tensor_array.cc b/paddle/framework/tensor_array.cc
@@ -254,13 +254,12 @@ LoDTensor TensorArray::LodPackTwo(const LoDTensor& pre, const LoDTensor& cur,
 void TensorArray::LodUnpack(const LoDTensor& source, size_t level) {
   PADDLE_ENFORCE_EQ(level, source.NumLevels() - 1,
                     "only the lowest LoD level supports unpack.");
-  int non_empty_instances = -1;
+  const size_t non_empty_instances = source.dims()[0];
   size_t index = 0;
   Vector<size_t> lowest_lod_level;
   lowest_lod_level.push_back(index);
 
-  for (size_t step = 0; non_empty_instances > 0 || non_empty_instances == -1;
-       step++) {
+  for (size_t step = 0; step < non_empty_instances; step++) {
     size_t num_instances = 0;
     for (size_t id = 0; id < source.NumElements(level); id++) {
       auto instance = source;