I am trying to modify data flow of a tensorflow model at runtime. Consider a 3 layers FC neural network. Let\'s say I want to define 2 different layers for the middle positi