diff --git a/torch_npu/csrc/framework/utils/KernelNpuOutputSize.cpp b/torch_npu/csrc/framework/utils/KernelNpuOutputSize.cpp
index 1de3c52de2e608111fa431b1f11432c31e4cc3ec..fe955737d3a646387fab11ebf8a20653b2b4ea13 100644
--- a/torch_npu/csrc/framework/utils/KernelNpuOutputSize.cpp
+++ b/torch_npu/csrc/framework/utils/KernelNpuOutputSize.cpp
@@ -347,6 +347,7 @@ namespace native {
             int64_t N = input.size(0);
             int64_t L = input.size(2);
             int64_t C_out = weight.size(0);
+            C_out = (weight.size(1) != 0) ? C_out : 0;
 
             auto kernel_size = weight.sizes().slice(2);