Fix setting lod

8 years ago · 1234b8b489
parent 7ccbc706bf
commit 1234b8b489
1 changed files with 1 additions and 1 deletions
--- a/paddle/operators/im2sequence_op.h
+++ b/paddle/operators/im2sequence_op.h
@ -79,7 +79,7 @@ class Im2SequenceKernel : public framework::OpKernel<T> {
    framework::LoD lod(1);
    lod[0].reserve(batch_size + 1);
    for (int i = 0, offset = 0; i < batch_size + 1; ++i) {
-      lod[0][i] = offset;
+      lod[0].push_back(offset);
      offset += output_height * output_width;
    }
    out->set_lod(lod);