https://jdhao.github.io/2017/11/12/pytorch-computation-graph/