Computing batch Jacobian efficiently

6

Note that if you’re using the latest version of pytorch, there is a vectorize=True flag for functional.jacobian() that might speed things up in some cases :slight_smile:

Calculate the Jacobian batch-wise of multiple input neural network with complex weights