A better way would be for the neural network to generate input parameters for Finite Element Method solvers.