add prelu

x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py

@@ -1359,6 +1359,34 @@ class OpSet9():
         elif len(shape_slope) > 2:
             raise Exception("The 'element' mode is not supported yet!")
 
+        if mode == "element":
+            self.paddle_graph.add_layer(
+                "paddle.zeros",
+                inputs={}, 
+                outputs=[output_name + "__zeros"], 
+                shape=shape_slope,
+                dtype=string(node.dtype))
+            self.paddle_graph.add_layer(
+                "paddle.maximum",
+                inputs={"x": val_x.name, 
+                        "y": output_name + "__zeros"}, 
+                outputs=[output_name + "__max"])
+            self.paddle_graph.add_layer(
+                "paddle.minimum",
+                inputs={"x": val_x.name, 
+                        "y": output_name + "__zeros"}, 
+                outputs=[output_name + "__max"])
+            self.paddle_graph.add_layer(
+                "paddle.multiply",
+                inputs={"x": val_slope.name, 
+                        "y": output_name + "__min"}, 
+                outputs=[output_name + "__mul"])
+            self.paddle_graph.add_layer(
+                "paddle.add",
+                inputs={"x": output_name + "__max", 
+                        "y": output_name + "__mul"}, 
+                outputs=[output_name])
+        else:
             if mode == 'channel' and len(shape_slope) == 1:
                 # paddle params shape need be [1, channel]
                 slope_data = _const_weight_or_none(val_slope)

x2paddle/op_mapper/static/onnx2paddle/opset9/opset.py

@@ -1336,6 +1336,14 @@ class OpSet9():
         elif len(shape_slope) > 2:
             raise Exception("The 'element' mode is not supported yet!")
 
+        if mode == "element":
+            self.paddle_graph.add_layer(
+                "paddle.static.nn.prelu", 
+                inputs={"x": val_x.name,
+                        "param_attr": val_slope.name}, 
+                outputs=[node.name],
+                mode="element")
+        else:
             if mode == 'channel' and len(shape_slope) == 1:
                 # paddle params shape need be [1, channel]
                 slope_data = _const_weight_or_none(val_slope)