See all of the possible distribution combinations that can be used in PyTorch's kl_divergence function.

view_torch_kl_options.py

from torch.distributions.kl import _KL_REGISTRY

def view_kl_options():
    """
    Displays all combinations of distributions that can be used in 
    torch's kl_divergence function. Iterates through the registry 
    and prints out the registered name combos.
    """
    names = [(k[0].__name__, k[1].__name__) for k in _KL_REGISTRY.keys()]
    max_name_len = max([len(t[0]) for t in names])
    for arg1, arg2 in sorted(names):
        print(f"  {arg1:>{max_name_len}} || {arg2}")