[API Compatibility] add torch.nn.ParameterDict -part

paconvert/api_mapping.json

@@ -8447,15 +8447,7 @@
     "Matcher": "ChangePrefixMatcher"
   },
   "torch.nn.ParameterDict": {
-    "Matcher": "GenericMatcher",
-    "paddle_api": "paddle.nn.ParameterDict",
-    "min_input_args": 0,
-    "args_list": [
-      "values"
-    ],
-    "kwargs_change": {
-      "values": "parameters"
-    }
+    "Matcher": "ChangePrefixMatcher"
   },
   "torch.nn.ParameterList": {
     "Matcher": "GenericMatcher",

tests/test_nn_ParameterDict.py

@@ -92,3 +92,164 @@ def test_case_5():
         """
     )
     obj.run(pytorch_code, ["result"])
+
+
+def test_case_6():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        choices = nn.ParameterDict(parameters={
+            'a': nn.Parameter(torch.ones(2, 3)),
+            'b': nn.Parameter(torch.zeros(4)),
+        })
+        result = list(choices)
+        """
+    )
+    obj.run(pytorch_code, ["result"])
+
+
+def test_case_7():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        choices = nn.ParameterDict([
+            ('a', nn.Parameter(torch.ones(2, 3))),
+            ('b', nn.Parameter(torch.zeros(4))),
+        ])
+        result = list(choices)
+        """
+    )
+    obj.run(pytorch_code, ["result"])
+
+
+def test_case_8():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        choices = nn.ParameterDict({'w': nn.Parameter(torch.ones(2, 3))})
+        result = choices['w']
+        """
+    )
+    obj.run(pytorch_code, ["result"], check_stop_gradient=False)
+
+
+def test_case_9():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        choices = nn.ParameterDict({
+            'a': nn.Parameter(torch.ones(1)),
+            'b': nn.Parameter(torch.ones(2)),
+            'c': nn.Parameter(torch.ones(3)),
+        })
+        result = len(choices)
+        """
+    )
+    obj.run(pytorch_code, ["result"])
+
+
+def test_case_10():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        choices = nn.ParameterDict({
+            'a': nn.Parameter(torch.ones(1)),
+            'b': nn.Parameter(torch.ones(2)),
+        })
+        result = list(choices.keys())
+        """
+    )
+    obj.run(pytorch_code, ["result"])
+
+
+def test_case_11():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        choices = nn.ParameterDict({
+            'a': nn.Parameter(torch.ones(2, 3)),
+            'b': nn.Parameter(torch.zeros(4)),
+        })
+        result = list(choices.values())
+        """
+    )
+    obj.run(pytorch_code, ["result"], check_stop_gradient=False)
+
+
+def test_case_12():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        choices = nn.ParameterDict({'a': nn.Parameter(torch.ones(2))})
+        result = 'a' in choices
+        """
+    )
+    obj.run(pytorch_code, ["result"])
+
+
+def test_case_14():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        choices = nn.ParameterDict(parameters={
+            'a': nn.Parameter(torch.ones(2, 3)),
+            'b': nn.Parameter(torch.zeros(4)),
+        })
+        result = choices['a']
+        """
+    )
+    obj.run(pytorch_code, ["result"], check_stop_gradient=False)
+
+
+def test_case_15():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        choices = nn.ParameterDict(parameters={
+            'a': nn.Parameter(torch.ones(2, 3)),
+            'b': nn.Parameter(torch.zeros(4)),
+        })
+        result = choices['b']
+        """
+    )
+    obj.run(pytorch_code, ["result"], check_stop_gradient=False)
+
+
+def test_case_13():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        choices = nn.ParameterDict({
+            'a': nn.Parameter(torch.ones(2, 3)),
+            'b': nn.Parameter(torch.zeros(4)),
+        })
+        result = choices.pop('a')
+        """
+    )
+    obj.run(pytorch_code, ["result"], check_stop_gradient=False)
+
+
+def test_case_16():
+    pytorch_code = textwrap.dedent(
+        """
+        import torch.nn as nn
+        import torch
+        pd1 = nn.ParameterDict({
+            'a': nn.Parameter(torch.ones(2, 3)),
+            'b': nn.Parameter(torch.zeros(4)),
+        })
+        pd2 = nn.ParameterDict(pd1)
+        result = list(pd2)
+        """
+    )
+    obj.run(pytorch_code, ["result"])