MIR.deformation_regularizer.KL_divergence

class MIR.deformation_regularizer.KL_divergence(*args, **kwargs)[source]

KL divergence between factorized (voxel- and component-wise independent) Gaussian fields.

Computes KL(P || Q) for univariate normals at every element and averages over all elements. The constructor takes no arguments.

__init__()[source]

Methods

__init__()

forward(P, Q)

Compute KL divergence between two Gaussian fields.