And, the vector here is the torch.ones_like(out)? and Jacobian is the derivative of the forward w.r.t whatever you’re differentiating? Is that correct?
Are there any textbooks (or other material) that would explain this is more finer detail?
Thank you for your help!