pytorch

1
ir_version: 8
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
5
  node {
6
    input: "onnx::Flatten_0"
7
    output: "1"
8
    name: "Flatten_0"
9
    op_type: "Flatten"
10
    attribute {
11
      name: "axis"
12
      i: 1
13
      type: INT
14
    }
15
  }
16
  name: "main_graph"
17
  input {
18
    name: "onnx::Flatten_0"
19
    type {
20
      tensor_type {
21
        elem_type: 1
22
        shape {
23
          dim {
24
            dim_value: 1
25
          }
26
          dim {
27
            dim_value: 2
28
          }
29
          dim {
30
            dim_value: 3
31
          }
32
          dim {
33
            dim_value: 4
34
          }
35
        }
36
      }
37
    }
38
  }
39
  output {
40
    name: "1"
41
    type {
42
      tensor_type {
43
        elem_type: 1
44
        shape {
45
          dim {
46
            dim_value: 1
47
          }
48
          dim {
49
            dim_value: 24
50
          }
51
        }
52
      }
53
    }
54
  }
55
}
56
opset_import {
57
  version: 17
58
}
59