fix Pad3dOpLite::InferShapeImpl() (#6824)

PaddlePaddle · Sep 6, 2021 · 6bf917a · 6bf917a
1 parent 80b25fa
commit 6bf917a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lite/operators/pad3d_op.cc b/lite/operators/pad3d_op.cc
@@ -42,7 +42,7 @@ bool Pad3dOpLite::InferShapeImpl() const {
     out_d = x_dims[1] + param_.paddings[4] + param_.paddings[5];
     out_h = x_dims[2] + param_.paddings[2] + param_.paddings[3];
     out_w = x_dims[3] + param_.paddings[0] + param_.paddings[1];
-    param_.Out->Resize(lite::DDim({x_dims[0], out_d, out_h, out_w, x_dims[1]}));
+    param_.Out->Resize(lite::DDim({x_dims[0], out_d, out_h, out_w, x_dims[4]}));
   } else {
     param_.Out->Resize(lite::DDim({x_dims[0], x_dims[1], out_d, out_h, out_w}));
   }