Axis attribute

justinchuby · Sep 22, 2023 · eb3526a · eb3526a
1 parent 662e46e
commit eb3526a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/onnx/test/automatic_upgrade_test.py b/onnx/test/automatic_upgrade_test.py
@@ -1546,6 +1546,9 @@ def test_DFT(self) -> None:
         self._test_op_upgrade(
             "DFT", 17, [[2, 16, 2], []], [[2, 16, 2]], attrs={"inverse": 1}
         )
+        self._test_op_upgrade(
+            "DFT", 17, [[2, 16, 2], []], [[2, 16, 2]], attrs={"inverse": 1, "axis": 0}
+        )
 
     def _test_short_time_fourier_transform(self, operator_name: str) -> None:
         # Real