register_kl¶

paddle.distribution. register_kl ( cls_p: type[Distribution], cls_q: type[Distribution] ) → Callable[[_T], _T] [source]

Decorator for register a KL divergence implementation function.

The kl_divergence(p, q) function will search concrete implementation functions registered by register_kl, according to multi-dispatch pattern. If an implementation function is found, it will return the result, otherwise, it will raise NotImplementError exception. Users can register implementation function by the decorator.

Parameters

• cls_p (type[Distribution]) – The Distribution type of Instance p. Subclass derived from Distribution.

• cls_q (type[Distribution]) – The Distribution type of Instance q. Subclass derived from Distribution.

Examples

>>> import paddle

>>> @paddle.distribution.register_kl(paddle.distribution.Beta, paddle.distribution.Beta)
>>> def kl_beta_beta():
...     pass # insert implementation here