Sorry, I didn’t explain it correctly. Yes, I need a full Jacobian with 6BB values. What is the most efficient way to compute it? I don’t want to use a double loop for all 6BB values and I assume that it can’t be computed directly done with a single backprop pass with torch.autograd. Ultimately, I want to compute L2 norm of this full Jacobian. If there’s a more efficient way to do this, let me know. I am still new to PyTorch and learning about it.