It seems that this network only works for the 2D or 1D images. If I want to solve a chemical ODE, does it work?