fix(onnx): 改正 Concat 形状推导中的笔误

Signed-off-by: YdrMaster <[email protected]>
InfiniTensor · Nov 13, 2023 · 636bf70 · 636bf70
1 parent 10712b5
commit 636bf70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/07onnx/src/operators/concat.cc b/src/07onnx/src/operators/concat.cc
@@ -40,7 +40,7 @@ namespace refactor::onnx {
                 return Err(InferError(ERROR_MSG("Input shape not support")));
             }
             for (auto i : range0_(static_cast<int64_t>(output.size()))) {
-                if (i == axis) {
+                if (i == axis_) {
                     EXPECT_VAL(output[i], a)
                     EXPECT_VAL(input.shape[i], b)
                     output[i] = DimExpr(a + b);