Solved by Adam_Conkey in post #3 
              
              
                Thank you for the suggestion @googlebot. Based on that I wrote this function that does what I want: 
from torch.distributions.kl import _KL_REGISTRY

def view_kl_options():
    names = [(k[0].__name__, k[1].__name__) for k in _KL_REGISTRY.keys()]
    max_name_len = max([len(t[0]) for t in names])
  …