Add test for avg_pool

Author: wodesuck

tests/function_libs/torch_lib/e2e_ops_tests.py

@@ -238,6 +238,30 @@ def forward(self, x):
         )
         _testing.assert_onnx_program(onnx_program)
 
+    def test_avg_pool(self):
+        class Model(torch.nn.Module):
+            def forward(self, x2d, x3d, x4d, x5d):
+                return (
+                    torch.nn.functional.avg_pool1d(x2d, 2),
+                    torch.nn.functional.avg_pool1d(x3d, 2),
+                    torch.nn.functional.avg_pool2d(x3d, 2),
+                    torch.nn.functional.avg_pool2d(x4d, 2),
+                    torch.nn.functional.avg_pool3d(x4d, 2),
+                    torch.nn.functional.avg_pool3d(x5d, 2),
+                )
+
+        x2d = torch.randn(10, 10)
+        x3d = torch.randn(10, 10, 10)
+        x4d = torch.randn(10, 10, 10, 10)
+        x5d = torch.randn(10, 10, 10, 10, 10)
+        onnx_program = torch.onnx.export(
+            Model(),
+            (x2d, x3d, x4d, x5d),
+            dynamo=True,
+            verbose=False,
+        )
+        _testing.assert_onnx_program(onnx_program)
+
 
 if __name__ == "__main__":
     unittest.main()