We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7ccbc70 commit 1234b8bCopy full SHA for 1234b8b
paddle/operators/im2sequence_op.h
@@ -79,7 +79,7 @@ class Im2SequenceKernel : public framework::OpKernel<T> {
79
framework::LoD lod(1);
80
lod[0].reserve(batch_size + 1);
81
for (int i = 0, offset = 0; i < batch_size + 1; ++i) {
82
- lod[0][i] = offset;
+ lod[0].push_back(offset);
83
offset += output_height * output_width;
84
}
85
out->set_lod(lod);
0 commit comments