Fasc:

As for my last comment the batch size n is confused for the input dimension, in this can 2. Does that make more sense?


Ho you mean you have a single argument that is a tuple? We can’t really handle that indeed. You can wrap it in a tuple before passing it to the hessian function for this to work: hessian(model, (my_tuple_arg,)).