Build a neural network which can work with 2D and 1D data such that I can use ground state and stiffness of cgdna.