Hi, do you mean the in-build function also works on one input point?
By the way, may I ask for an example of https://pytorch.org/docs/stable/autograd.html#torch.autograd.functional.jacobian 10 with respect to the parameters of a network please? I have no clue since the first parameter of jacobian is a function.