From c2eec47b42c70b96f904e7c381dd4cb61baa65d5 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Sat, 9 Apr 2022 01:08:50 +0800
Subject: [PATCH] fix(imperative/src): fix adaptive_pooling bug

GitOrigin-RevId: e1dc46e3fa65562077f91bf4c18df536d53f322b
---
 imperative/src/impl/ops/adaptive_pooling.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/imperative/src/impl/ops/adaptive_pooling.cpp b/imperative/src/impl/ops/adaptive_pooling.cpp
index c8e4fcae..0521ce1d 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) {