1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
# Sample UnPack model, axis = 1
model = Model()
input = Input("input", "TENSOR_FLOAT32", "{6, 3, 4}")
axis = Int32Scalar("axis", 1)
num_splits = Int32Scalar("num_splits", 3)
out1 = Output("output1", "TENSOR_FLOAT32", "{6, 4}")
out2 = Output("output2", "TENSOR_FLOAT32", "{6, 4}")
out3 = Output("output3", "TENSOR_FLOAT32", "{6, 4}")
model = model.Operation("UNPACK_EX", input, num_splits, axis).To([out1, out2, out3])
input0 = {input: # input 0
[0.3, 1.2, 2.1, 3.454,
4.12, 5.23, 6.34, 7.12,
8.2, 9.23, 10.11, 11.34,
12.3, 13.12, 14.122, 15.1,
16.11, 17.234, 18.2, 19.1,
20.45, 21.5, 22.5, 23.64,
24.2, 25.76, 26.4, 27.34,
28.43, 29.112, 30.45, 31.5,
32.45, 33.65, 34.6, 35.34,
36.3, 37.54, 38.6743, 39.56,
40.451, 41.56, 42.55, 43.511,
44.4, 45.45, 46.5641, 47.122,
48.1, 49.456, 50.56, 51.11,
52.1, 53.34, 54.62, 55.1,
56.12, 57.4, 58.2, 59.1,
60.465, 61.1, 62.3, 63.45,
64.1, 65.11, 66.4, 67.9,
68.123, 69.65, 70.89, 71.987]}
output0 = {out1: # output 0
[0.3, 1.2, 2.1, 3.454,
12.3, 13.12, 14.122, 15.1,
24.2, 25.76, 26.4, 27.34,
36.3, 37.54, 38.6743, 39.56,
48.1, 49.456, 50.56, 51.11,
60.465, 61.1, 62.3, 63.45],
out2: # output 1
[4.12, 5.23, 6.34, 7.12,
16.11, 17.234, 18.2, 19.1,
28.43, 29.112, 30.45, 31.5,
40.451, 41.56, 42.55, 43.511,
52.1, 53.34, 54.62, 55.1,
64.1, 65.11, 66.4, 67.9],
out3: # output 2
[8.2, 9.23, 10.11, 11.34,
20.45, 21.5, 22.5, 23.64,
32.45, 33.65, 34.6, 35.34,
44.4, 45.45, 46.5641, 47.122,
56.12, 57.4, 58.2, 59.1,
68.123, 69.65, 70.89, 71.987]}
# Instantiate an example
Example((input0, output0))
|
