https://oldpan.me/archives/pytorch-autograd-hook#https://www.freesion.com/article/41721100168/https://www.cnblogs.com/hellcat/p/8512090.html
