1 files changed, 18 insertions, 0 deletions
diff --git a/res/PyTorchExamples/examples/AvgPool2d/__init__.py b/res/PyTorchExamples/examples/AvgPool2d/__init__.py
new file mode 100644
index 000000000..f26521f21
--- /dev/null
+++ b/res/PyTorchExamples/examples/AvgPool2d/__init__.py
@@ -0,0 +1,18 @@
+import torch
+import torch.nn as nn
+
+
+# model
+class net_AvgPool2d(nn.Module):
+    def __init__(self):
+        super().__init__()
+        self.op = nn.AvgPool2d(1)
+
+    def forward(self, input):
+        return self.op(input)
+
+
+_model_ = net_AvgPool2d()
+
+# dummy input for onnx generation
+_dummy_ = torch.randn(1, 2, 3, 3)