【Hackathon 7th No.36】为 Paddle 代码转换工具新增 API 转换规则（第 3 组）

paconvert/api_mapping.json

@@ -15024,6 +15024,202 @@
       "input": "x"
     }
   },
+  "torch.signal.windows.blackman": {
+    "Matcher": "SignalWindowsWatcher",
+    "paddle_api": "paddle.audio.functional.get_window",
+    "min_input_args": 1,
+    "args_list": [
+      "M",
+      "*",
+      "sym",
+      "dtype",
+      "layout",
+      "device",
+      "requires_grad"
+    ],
+    "kwargs_change": {
+      "M": "win_length",
+      "sym": "fftbins",
+      "dtype": "dtype"
+    },
+    "paddle_default_kwargs": {
+      "dtype": "'float32'",
+      "fftbins": "False",
+      "window": "'blackman'"
+    }
+  },
+  "torch.signal.windows.cosine": {
+    "Matcher": "SignalWindowsWatcher",
+    "paddle_api": "paddle.audio.functional.get_window",
+    "min_input_args": 1,
+    "args_list": [
+      "M",
+      "*",
+      "sym",
+      "dtype",
+      "layout",
+      "device",
+      "requires_grad"
+    ],
+    "kwargs_change": {
+      "M": "win_length",
+      "sym": "fftbins",
+      "dtype": "dtype"
+    },
+    "paddle_default_kwargs": {
+      "dtype": "'float32'",
+      "fftbins": "False",
+      "window": "'cosine'"
+    }
+  },
+  "torch.signal.windows.exponential": {
+    "Matcher": "SignalWindowsWatcher",
+    "paddle_api": "paddle.audio.functional.get_window",
+    "min_input_args": 1,
+    "args_list": [
+      "M",
+      "*",
+      "center",
+      "tau",
+      "sym",
+      "dtype",
+      "layout",
+      "device",
+      "requires_grad"
+    ],
+    "unsupport_args": [
+      "center"
+    ],
+    "kwargs_change": {
+      "M": "win_length",
+      "sym": "fftbins",
+      "dtype": "dtype"
+    },
+    "paddle_default_kwargs": {
+      "dtype": "'float32'",
+      "fftbins": "False"
+    }
+  },
+  "torch.signal.windows.gaussian": {
+    "Matcher": "SignalWindowsWatcher",
+    "paddle_api": "paddle.audio.functional.get_window",
+    "min_input_args": 1,
+    "args_list": [
+      "M",
+      "*",
+      "std",
+      "sym",
+      "dtype",
+      "layout",
+      "device",
+      "requires_grad"
+    ],
+    "kwargs_change": {
+      "M": "win_length",
+      "sym": "fftbins",
+      "dtype": "dtype"
+    },
+    "paddle_default_kwargs": {
+      "dtype": "'float32'",
+      "fftbins": "False"
+    }
+  },
+  "torch.signal.windows.general_cosine": {
+    "Matcher": "SignalWindowsWatcher",
+    "paddle_api": "paddle.audio.functional.get_window",
+    "min_input_args": 1,
+    "args_list": [
+      "M",
+      "*",
+      "a",
+      "sym",
+      "dtype",
+      "layout",
+      "device",
+      "requires_grad"
+    ],
+    "kwargs_change": {
+      "M": "win_length",
+      "sym": "fftbins",
+      "dtype": "dtype"
+    },
+    "paddle_default_kwargs": {
+      "dtype": "'float32'",
+      "fftbins": "False"
+    }
+  },
+  "torch.signal.windows.general_hamming": {
+    "Matcher": "SignalWindowsWatcher",
+    "paddle_api": "paddle.audio.functional.get_window",
+    "min_input_args": 1,
+    "args_list": [
+      "M",
+      "*",
+      "alpha",
+      "sym",
+      "dtype",
+      "layout",
+      "device",
+      "requires_grad"
+    ],
+    "kwargs_change": {
+      "M": "win_length",
+      "sym": "fftbins",
+      "dtype": "dtype"
+    },
+    "paddle_default_kwargs": {
+      "dtype": "'float32'",
+      "fftbins": "False"
+    }
+  },
+  "torch.signal.windows.hamming": {
+    "Matcher": "SignalWindowsWatcher",
+    "paddle_api": "paddle.audio.functional.get_window",
+    "min_input_args": 1,
+    "args_list": [
+      "M",
+      "*",
+      "sym",
+      "dtype",
+      "layout",
+      "device",
+      "requires_grad"
+    ],
+    "kwargs_change": {
+      "M": "win_length",
+      "sym": "fftbins",
+      "dtype": "dtype"
+    },
+    "paddle_default_kwargs": {
+      "dtype": "'float32'",
+      "fftbins": "False",
+      "window": "'hamming'"
+    }
+  },
+  "torch.signal.windows.hann": {
+    "Matcher": "SignalWindowsWatcher",
+    "paddle_api": "paddle.audio.functional.get_window",
+    "min_input_args": 1,
+    "args_list": [
+      "M",
+      "*",
+      "sym",
+      "dtype",
+      "layout",
+      "device",
+      "requires_grad"
+    ],
+    "kwargs_change": {
+      "M": "win_length",
+      "sym": "fftbins",
+      "dtype": "dtype"
+    },
+    "paddle_default_kwargs": {
+      "dtype": "'float32'",
+      "fftbins": "False",
+      "window": "'hann'"
+    }
+  },
   "torch.signbit": {
     "Matcher": "GenericMatcher",
     "paddle_api": "paddle.signbit",

paconvert/api_matcher.py

@@ -519,6 +519,36 @@ def generate_code(self, kwargs):
         return super().generate_code(kwargs)
 
 
+class SignalWindowsWatcher(BaseMatcher):
+    def generate_code(self, kwargs):
+        new_kwargs = {}
+        if "sym" in kwargs:
+            kwargs["fftbins"] = "not " + kwargs.pop("sym")
+        if "exponential" in self.torch_api:
+            if "tau" in kwargs:
+                new_kwargs["window"] = "('exponential', None, {})".format(
+                    kwargs.pop("tau")
+                )
+            else:
+                new_kwargs["window"] = "('exponential', None, 1.0)"
+        if "gaussian" in self.torch_api:
+            if "std" in kwargs:
+                new_kwargs["window"] = "('gaussian', {})".format(kwargs.pop("std"))
+            else:
+                new_kwargs["window"] = "('gaussian', 1.0)"
+        if "general_hamming" in self.torch_api:
+            if "alpha" in kwargs:
+                new_kwargs["window"] = "('general_hamming', {})".format(
+                    kwargs.pop("alpha")
+                )
+            else:
+                new_kwargs["window"] = "('general_hamming', 0.54)"
+        if "general_cosine" in self.torch_api:
+            new_kwargs["window"] = "('general_cosine', {})".format(kwargs.pop("a"))
+        new_kwargs.update(kwargs)
+        return GenericMatcher.generate_code(self, new_kwargs)
+
+
 class Num2TensorBinaryWithAlphaMatcher(BaseMatcher):
     def generate_code(self, kwargs):
         kwargs_change = self.api_mapping.get("kwargs_change", {})
@@ -832,7 +862,7 @@ def get_paddle_nodes(self, args, kwargs):
             else:
                 shape = self.parse_args(args)[0]
             kwargs = {"shape": str(shape).replace("'", ""), **kwargs}
-        
+
         if "dtype" not in kwargs:
             kwargs["dtype"] = "float32"
 
@@ -841,7 +871,7 @@ def get_paddle_nodes(self, args, kwargs):
 
         if "high" not in kwargs:
             kwargs["high"] = 1
-        
+
         if "requires_grad" not in kwargs.keys():
             API_TEMPLATE = textwrap.dedent(
                 """
@@ -871,7 +901,7 @@ def get_paddle_nodes(self, args, kwargs):
                 kwargs["device"],
                 kwargs["requires_grad"],
             )
-        
+
         return ast.parse(code).body