How to express partial dirivates of a 2-argument function in TensorFlow

Question

I want to solve a partial differential equation (PDE) with TensorFlow, which is a 2-argument function, the two arguments are t(time) and e(energy). I construct a neural netw