Vit operators

• Added Erf, Concat, Gather, ReduceMean, Reshape, Slice and transpose operators and their python bindings
• Modified Softmax operator by adding axis attribute to support import from ONNX
• Added RemoveDropout Recipe to remove dropout nodes from graphs

aidge_core/unit_tests/test_recipies.py

@@ -20,6 +20,18 @@ class test_recipies(unittest.TestCase):
     def tearDown(self):
         pass
 
+    def test_remove_dropout(self):
+        graph_view = aidge_core.sequential([
+            aidge_core.GenericOperator("Conv", 1, 0, 1, "Conv0"),
+            aidge_core.GenericOperator("Dropout", 1, 0, 1, name="Dropout0")
+        ])
+        old_nodes = graph_view.get_nodes()
+        aidge_core.remove_dropout(graph_view)
+        self.assertTrue(len(graph_view.get_nodes()) == len(old_nodes) - 1)
+        self.assertTrue("Dropout0" not in [i.name for i in graph_view.get_nodes()])
+
+        self.assertTrue(all([i in old_nodes for i in graph_view.get_nodes()]))
+
     def test_remove_flatten(self):
         graph_view = aidge_core.sequential([
             aidge_core.GenericOperator("Flatten", 1, 0, 1, name="Flatten0"),