torch.distributions.register_kl
function register_kl(p_type: string, q_type: string, fn: KLFunction): voidRegister a KL divergence function for a pair of distribution types.
Parameters
p_typestringq_typestringfnKLFunction
Examples
register_kl('Normal', 'Normal', (p, q) => {
// Compute KL(p || q) for Normal distributions
...
});