diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/concat_cpu_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/concat_cpu_kernel.cc index 33852c8536..5d73a25bff 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/concat_cpu_kernel.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/concat_cpu_kernel.cc @@ -53,6 +53,9 @@ bool ConcatCPUKernel::Launch(const std::vector &inputs, auto before_axis = input_flat_shape_list[0][0]; for (size_t i = 0; i < before_axis; ++i) { for (size_t j = 0; j < input_num; ++j) { + if (input_flat_shape_list[j][1] == 0) { + continue; + } auto input_j_addr = reinterpret_cast(inputs[j]->addr); auto copy_num = input_flat_shape_list[j][1]; auto offset = copy_num * i;