Merge pull request #7811 from wanghaoshuang/fix_im2seq

wanghaoshuang · web-flow · commit 42549fb79211 · 2018-01-24T13:28:59.000+08:00
Fix LoD calculate of im2seq
diff --git 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);

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,7 @@ class Im2SequenceKernel : public framework::OpKernel<T> {`
`79`	`79`	`framework::LoD lod(1);`
`80`	`80`	`lod[0].reserve(batch_size + 1);`
`81`	`81`	`for (int i = 0, offset = 0; i < batch_size + 1; ++i) {`
`82`		`- lod[0][i] = offset;`
	`82`	`+ lod[0].push_back(offset);`
`83`	`83`	`offset += output_height * output_width;`
`84`	`84`	`}`
`85`	`85`	`out->set_lod(lod);`