register_kl

paddle.distribution. register_kl ( cls_p, cls_q ) [源代码]

用于注册KL散度具体计算函数装饰器。

调用 kl_divergence(p,q) 计算KL散度时，会通过多重派发机制，即根据p和q的类型查找通过 register_kl 注册的实现函数，如果找到返回计算结果，否则，抛出 NotImplementError 。 用户可通过该装饰器自行注册KL散度计算函数。

参数

  • cls_p (Distribution) - 实例p的分布类型，继承于Distribution基类。

  • cls_q (Distribution) - 实例q的分布类型，继承于Distribution基类。

代码示例

import paddle

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