diff --git a/imperative/src/impl/ops/broadcast.cpp b/imperative/src/impl/ops/broadcast.cpp
index ff35cfed98679a3ad7c03faa5cc8b68e5c990ae9..d2cb2c4b36833136261043d1f6eca09f069bd566 100644
--- a/imperative/src/impl/ops/broadcast.cpp
+++ b/imperative/src/impl/ops/broadcast.cpp
@@ -290,7 +290,6 @@ std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible(
         }
     }
     if (op.axis != opr::Reshape::Param::INVALID_AXIS) {
-        mgb_assert(out_shape[op.axis] == -1);
         out_shape[op.axis] = 1;
         mgb_assert(
                 src.layout.total_nr_elems() % out_shape.total_nr_elems() == 0,