fix(imperative/src): fix adaptive_pooling bug

GitOrigin-RevId: e1dc46e3fa65562077f91bf4c18df536d53f322b

fix(imperative/src): fix adaptive_pooling bug
GitOrigin-RevId: e1dc46e3fa65562077f91bf4c18df536d53f322b
c2eec47b · Megvii Engine Team · ffbf8fad · c2eec47b
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

imperative/src/impl/ops/adaptive_pooling.cpp imperative/src/impl/ops/adaptive_pooling.cpp +1 -0

未找到文件。
--- 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) {