Hi ptrblck, exactly, I will have to slice the input and feed to networks respectively. However, after slicing the input, I will get two parts, one for mlp1() and the other for mlp2(). The tricky part is that I need to merge the two parts back into a holistic tensor after forwarding the network. Any hints?