Follow up,
what if input_type is a control flow. e.g. a tensor: 2 x 5: batch number = 2, each vector contains 5 variables. For example: [1,1,0,0,1]. How to implement this structure?
For example:
[[1,1,0,0,1],[0,1,0,1,0]]
The control flow decide which layer to go, but different sequence has different control flow.