add test

pytorch · Jan 28, 2021 · 3ba0370 · 3ba0370
1 parent 9e7f1ac
commit 3ba0370
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/test/onnx/test_pytorch_onnx_onnxruntime.py b/test/onnx/test_pytorch_onnx_onnxruntime.py
@@ -4292,6 +4292,13 @@ def forward(self, x):
         x = torch.randn(4, 2, 4, requires_grad=True)
         self.run_test(UnfoldModel(), x)
 
+        class UnfoldModel(torch.nn.Module):
+            def forward(self, x):
+                return x.unfold(dimension=2, size=x.shape[1], step=1)
+
+        x = torch.randn(4, 2, 4, requires_grad=True)
+        self.run_test(UnfoldModel(), x)
+
     def test_prelu(self):
         class PReluModel(torch.nn.Module):
             def __init__(self):