priors.TotalVariation#

class bartorch.priors.TotalVariation#

Bases: Regularizer

Isotropic total variation, \(\lambda \sum_r \| (\nabla x)_r \|_2\) (BART’s T).

\(G = \nabla\) stacks the first differences along axes; the \(\ell_2\) norm is over the derivative directions.

Parameters:

axes (int or tuple of int) – Axes to differentiate, as indices into the image’s shape.
weight (float) – \(\lambda\).
joint_axes (int or tuple of int, default=()) – Axes along which entries are thresholded jointly.

Also has the methods and properties of Regularizer.

Examples using `TotalVariation`#