Representation of solving a partial differential equation using machine learning