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] Would anyone please guide me how to do this in Julia (Flux)?

How to find jacobian of learnt neural network model

ChrisRackauckas June 19, 2022, 4:09pm 2

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

Topic		Replies	Views
Autograd in Flux General Usage	14	1352	March 5, 2021
Gradient of jacobian in Flux v0.10 (zygote) General Usage machine-learning	0	430	February 13, 2020
Gradient of Flux model wrt to weights Machine Learning flux	4	1569	May 19, 2021
Gradient of NN with respect to inputs Machine Learning	1	374	January 8, 2021
How to use gradient of neural network as the loss function? Machine Learning question	13	2806	March 23, 2021