Total Variation (TV) Module Interface torchmetrics.image.TotalVariation Functional Interface torchmetrics.functional.image.total_variation