[BUG] Reproduce bug for fold with padding=1 (#30)

f-dangel · Jan 10, 2024 · bdd0057 · bdd0057
1 parent fbee3a4
commit bdd0057
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/fold_settings.py b/test/fold_settings.py
@@ -16,6 +16,16 @@
             "kernel_size": (2, 2),
         },
     },
+    {
+        "seed": 0,
+        "input_fn": lambda: torch.rand(2, 3 * 2 * 2, 5 * 9),
+        "fold_kwargs": {
+            "output_size": (4, 8),
+            "kernel_size": 2,
+            "padding": 1,
+        },
+        "id": "bug30-fold-with-padding",
+    },
 ]
 PROBLEMS_2D_IDS = [make_id(problem) for problem in PROBLEMS_2D]