# model
model = Model()
-i1 = Input("op1", "TENSOR_FLOAT32", "{4}") # a vector of input
-i2 = Output("op2", "TENSOR_FLOAT32", "{4}") # a vector of output
+i1 = Input("op1", "TENSOR_FLOAT32", "{4}") # a vector of 4 float32s
+i2 = Output("op2", "TENSOR_FLOAT32", "{4}") # a vector of 4 float32s
model = model.Operation("EXP_EX", i1).To(i2)
# Example 1. Input in operand 0,
# model
model = Model()
-i1 = Input("op1", "TENSOR_FLOAT32", "{3,4}") # a vector of 2 float32s
-i2 = Output("op2", "TENSOR_FLOAT32", "{3,4}")
+i1 = Input("op1", "TENSOR_FLOAT32", "{3,4}") # a vector of 12 float32s
+i2 = Output("op2", "TENSOR_FLOAT32", "{3,4}") # a vector of 12 float32s
model = model.Operation("EXP_EX", i1).To(i2)
# Example 1. Input in operand 0,
# model
model = Model()
-i1 = Input("op1", "TENSOR_FLOAT32", "{4}") # a vector of 2 float32s
+i1 = Input("op1", "TENSOR_FLOAT32", "{4}") # a vector of 4 float32s
i2 = Input("op2", "TENSOR_INT32", "{2}") # another vector of 2 int32s
axis = Int32Scalar("axis", 0)
-i3 = Output("op3", "TENSOR_FLOAT32", "{2}")
+i3 = Output("op3", "TENSOR_FLOAT32", "{2}") # a vector of 2 float32s
model = model.Operation("GATHER_EX", i1, i2, axis).To(i3)
# Example 1. Input in operand 0,
# model
model = Model()
-i1 = Input("op1", "TENSOR_FLOAT32", "{3,4}") # a vector of 2 float32s
+i1 = Input("op1", "TENSOR_FLOAT32", "{3,4}") # a vector of 12 float32s
i2 = Input("op2", "TENSOR_INT32", "{2}") # another vector of 2 int32s
axis = Int32Scalar("axis", 0)
-i3 = Output("op3", "TENSOR_FLOAT32", "{2,4}")
+i3 = Output("op3", "TENSOR_FLOAT32", "{2,4}") # a vector of 8 float32s
model = model.Operation("GATHER_EX", i1, i2, axis).To(i3)
# Example 1. Input in operand 0,