1 model = Model() 2 i1 = Input("input", "TENSOR_FLOAT32", "{4, 1, 1, 2}") 3 block = Parameter("block_size", "TENSOR_INT32", "{2}", [2, 2]) 4 output = Output("output", "TENSOR_FLOAT32", "{1, 2, 2, 2}") 5 6 model = model.Operation("BATCH_TO_SPACE_ND", i1, block).To(output) 7 8 # Example 1. Input in operand 0, 9 input0 = {i1: # input 0 10 [1.4, 2.3, 3.2, 4.1, 5.4, 6.3, 7.2, 8.1]} 11 12 output0 = {output: # output 0 13 [1.4, 2.3, 3.2, 4.1, 5.4, 6.3, 7.2, 8.1]} 14 15 # Instantiate an example 16 Example((input0, output0)) 17
