diff --git a/imperative/src/impl/ops/adaptive_pooling.cpp b/imperative/src/impl/ops/adaptive_pooling.cpp
index c8e4fcae1ac59380e939a110adbb9c2ad43c8325..0521ce1d57a54c2956f2771f0784c770101d3898 100644
--- a/imperative/src/impl/ops/adaptive_pooling.cpp
+++ b/imperative/src/impl/ops/adaptive_pooling.cpp
@@ -73,6 +73,7 @@ SmallVector<TensorPtr> apply_on_physical_tensor(
     TensorLayout dst_layout = output_descs[0].layout;
     if (!validated) {
         TensorShape tshp;
+        dst_layout.ndim = src_layout.ndim;
         dst_layout[0] = src_layout[0];
         dst_layout[1] = src_layout[1];
         if (inputs.size() == 2) {