fix(mge): fix dimshuffle shape infer

GitOrigin-RevId: ad7a73fbd1670aad926772d58b6e31fe52eeb448

imperative/src/impl/ops/specializations.cpp

@@ -81,6 +81,7 @@ std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible(
             src.layout.ndim);
     size_t idx = 0;
     bool input_used[TensorLayout::MAX_NDIM] = {0};
+    out_shape.ndim = ds.pattern.size();
     for (auto i : ds.pattern) {
         if (i < 0) {
             out_shape[idx] = 1;