diff --git a/cpp-package/example/feature_extract/feature_extract.cpp b/cpp-package/example/feature_extract/feature_extract.cpp
index c23623e23bd1..d614fd576238 100644
--- a/cpp-package/example/feature_extract/feature_extract.cpp
+++ b/cpp-package/example/feature_extract/feature_extract.cpp
@@ -106,6 +106,7 @@ class FeatureExtractor {
     /*print out the features*/
     auto array = executor->outputs[0].Copy(Context(kCPU, 0));
     NDArray::WaitAll();
+    array = array.Reshape({2, 1024});
     for (int i = 0; i < 1024; ++i) {
       cout << array.At(0, i) << ",";
     }