def _log_prob(self, given):
alpha, beta = self.alpha, self.beta
log_given = tf.log(given)
log_beta = tf.log(beta)
lgamma_alpha = tf.lgamma(alpha)
if self._check_numerics:
log_given = tf.check_numerics(log_given, "log(given)")
log_beta = tf.check_numerics(log_beta, "log(beta)")
lgamma_alpha = tf.check_numerics(lgamma_alpha, "lgamma(alpha)")
return alpha * log_beta - lgamma_alpha - (alpha + 1) * log_given - \
beta / given
评论列表
文章目录