My data for training neural network model contains current variables and also history variables to find best fit for non-linear function. Training perfomance is very good and model can predict data with very low error.
My main goal is to find tangent for the learnt model. I am using
f(array of input parameters both current and history variables) = NN_model
Zygote.jacobian(f,[ array of input parameters both current and history variables])
function to find tangent. This does not give results as expected. I want to find jacobian with respect to current variables only.
Zygote.jacobian(f,[ array of only parameters current]) → this gives error
Would it be possible with Zygote package or any other recommendations please?