modify onnx

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

@@ -1293,25 +1293,23 @@ class OpSet9():
         if shape_slope == [1]:
             mode = 'all'
         elif len(shape_slope) > 2:
-            mode = 'element'
+            raise Exception("The 'element' mode is not supported yet!")
 
         if mode == 'channel' and len(shape_slope) == 1:
             # paddle params shape need be [1, channel]
             slope_data = _const_weight_or_none(val_slope)
             slope_data = np.reshape(slope_data, [1] + shape_slope)
             self.weights[val_slope.layer_name] = slope_data
+            num_parameters = val_x.out_shapes[0][1]
+        else:
+            num_parameters = 1
 
-        layer_attrs = {
-            "param_attr": string(val_slope.layer_name),
-            'mode': string(mode),
-            "channel": val_x.out_shapes[0][1] if mode == "channel" else None,
-            "input_shape": val_x.out_shapes[0] if mode == "element" else None,
-        }
         self.paddle_graph.add_layer(
             "paddle.nn.PReLU", 
             inputs={"x": self.get_node_name(val_x)}, 
             outputs=layer_outputs, 
-            **layer_attrs)
+            num_parameters=num_parameters,
+            weight_attr=string(val_slope.layer_name))
 
     @print_mapping_info
     def Squeeze(self, node):