From 698ed7e3d9d4f4daba994e37edf0308bd710e0f3 Mon Sep 17 00:00:00 2001
From: ooooo <3164076421@qq.com>
Date: Tue, 10 Jun 2025 13:13:22 +0800
Subject: [PATCH] fix prod

---
 tester/paddle_to_torch/rules.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tester/paddle_to_torch/rules.py b/tester/paddle_to_torch/rules.py
index bf821da5..0dd4398a 100644
--- a/tester/paddle_to_torch/rules.py
+++ b/tester/paddle_to_torch/rules.py
@@ -4822,6 +4822,8 @@ def apply(self, paddle_api: str) -> ConvertResult:
     result = x
 """
             post = """
+if axis is None and keepdim:
+    result = result.view([1] * x.dim())
 if isinstance(axis, tuple) and not keepdim:
     result = torch.squeeze(result, dim=axis)
 """