diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py index 1a9b7993e4656dbbe901727db148b7be247b2b46..f4d461edcb631422aeb17c0e0eb6b40ec9667447 100644 --- a/mace/python/tools/converter_tool/transformer.py +++ b/mace/python/tools/converter_tool/transformer.py @@ -356,6 +356,13 @@ class Transformer(base_converter.ConverterInterface): self.safe_remove_node(op, self._producer.get(op.input[0], None)) return True + if op.type == 'Reshape' and \ + op.output_shape[0].dims == \ + self.get_tensor_shape(op.input[0]): + print("Remove useless reshape: %s(%s)" % (op.name, op.type)) + self.safe_remove_node(op, + self._producer.get(op.input[0], None)) + return True return False