How to find jacobian of learnt neural network model

Specific Domains Machine Learning
1

I modelled a neural network with a vector as input and a vector as output (multi input multi output NN model). Now I want to find jacobian of the learnt function as:

gradient
gradient978×193 22.5 KB

Would anyone please guide me how to do this in Julia (Flux)?

2

That’s not a gradient, that’s the Jacobian of f. See Zygote.jacobian.