Pytorch Total Variation