https://mobhitsbestcheapprice.blogspot.com/2024/05/understanding-jacobian-tensor-gradients.html
Understanding Jacobian Tensor Gradients In Pytorch