Calculate gradient of trained network

Question

assume I got a trained model, I am trying to calculate it\'s Jacobin (trying to understand some it\'s mathematical properties after training). I am trying to use autograd<