fix crash when input type is lod-tensor and input dims is 2 (#1703)

src/operators/feed_op.cpp

@@ -23,11 +23,7 @@ void FeedOp<DeviceType, T>::InferShape() const {
   out_dims[0] = this->param_.BatchSize();
   int col = this->param_.Col();
   auto input_dims = this->param_.InputX()->at(col).dims();
-  if (input_dims.size() == 4) {
-    this->param_.Out()->Resize(input_dims);
-  } else {
-    this->param_.Out()->Resize(out_dims);
-  }
+  this->param_.Out()->Resize(input_dims);
 }
 
 }  // namespace operators