remove transpose

x2paddle/op_mapper/static/tf2paddle/tf_op_mapper.py

@@ -96,6 +96,7 @@ class TFOpMapper(OpMapper):
             raise Exception("Model is not supported yet.")
         self.params = dict()
         self.paddle_graph = PaddleGraph(parent_layer=None, graph_type="static", source_type="tf")
+        self.params_output2id = dict()
 
         not_placeholder = list()
         for name in self.graph.input_nodes:
@@ -224,7 +225,7 @@ class TFOpMapper(OpMapper):
             return
 
         self.params[node.name] = node.value
-        self.paddle_graph.add_layer(
+        layer_id = self.paddle_graph.add_layer(
             kernel="paddle.static.create_parameter",
             inputs={},
             outputs=[node.name],
@@ -232,6 +233,7 @@ class TFOpMapper(OpMapper):
             shape=shape,
             name=string(node.name),
             default_initializer="paddle.nn.initializer.Constant(value=0.0)")
+        self.params_output2id[node.name] = layer_id
 
     def Transpose(self, node):
         input = self.graph.get_node(node.layer.input[0])
@@ -774,6 +776,12 @@ class TFOpMapper(OpMapper):
         data_format = node.get_attr("data_format").decode()
         pad_mode = node.get_attr("padding").decode()
 
+        if len(kernel.outputs) == 1:
+            self.params[kernel.name] = numpy.transpose(self.params[kernel.name],
+                                                          (2, 3, 0, 1))
+            layer = self.paddle_graph.layers[self.params_output2id[kernel.name]] 
+            layer.attrs["shape"] = self.params[kernel.name].shape
+        else:
             self.paddle_graph.add_layer(
                 kernel="paddle.transpose",
                 inputs={"x": kernel.name},