Fix

co63oc · co63oc · commit c190be1fcac6 · 2025-06-06T12:15:07.000+08:00
diff --git a/paddle/phi/kernels/impl/multi_dot_kernel_impl.h b/paddle/phi/kernels/impl/multi_dot_kernel_impl.h
@@ -187,6 +187,11 @@ void MultiDotKernel(const Context& ctx,
     if (x[i]->numel() == 0) size_0 = true;
   }
   if (size_0) {
+    // For example: [2, 0], [0, 4] -> [2, 4]
+    if (out && out->numel() > 0) {
+      phi::Full<T, Context>(
+          ctx, phi::IntArray(common::vectorize(out->dims())), 0, out);
+    }
     return;
   }
   const T scale = static_cast<T>(1.0);

Original file line number	Diff line number	Diff line change
`@@ -187,6 +187,11 @@ void MultiDotKernel(const Context& ctx,`
`187`	`187`	`if (x[i]->numel() == 0) size_0 = true;`
`188`	`188`	`}`
`189`	`189`	`if (size_0) {`
	`190`	`+ // For example: [2, 0], [0, 4] -> [2, 4]`
	`191`	`+ if (out && out->numel() > 0) {`
	`192`	`+ phi::Full<T, Context>(`
	`193`	`+ ctx, phi::IntArray(common::vectorize(out->dims())), 0, out);`
	`194`	`+ }`
`190`	`195`	`return;`
`191`	`196`	`}`
`192`	`197`	`const T scale = static_cast<T>(1.0);`