2d Heat Equation Pinn