I\'m trying to approximate a solution to a PDE and for that I need to compute 1st and 2nd order derivatives with respect to specific input values in my training data batch.<