ir_version: 8 producer_name: "pytorch" producer_version: "CURRENT_VERSION" graph { node { input: "onnx::Flatten_0" output: "1" name: "Flatten_0" op_type: "Flatten" attribute { name: "axis" i: 1 type: INT } } name: "main_graph" input { name: "onnx::Flatten_0" type { tensor_type { elem_type: 1 shape { dim { dim_value: 1 } dim { dim_value: 2 } dim { dim_value: 3 } dim { dim_value: 4 } } } } } output { name: "1" type { tensor_type { elem_type: 1 shape { dim { dim_value: 1 } dim { dim_value: 24 } } } } } } opset_import { version: 17 }