Change hessian generator function to only yield input and loss

[rostyhn]

We could make calculating the Hessian even more user friendly by only requiring the user to create a generator function that yields the loss and the input instead of returning the input size and gradient.