def normal_div_partial(self, pmean, plogvar, n):
prop = self.shape[0] / n
regul = prop * tf.mul(self.prec, tf.reduce_sum(tf.square(pmean), 0) + tf.reduce_sum(tf.exp(plogvar), 0))
return (tf.reduce_sum(regul) / 2.0
- self.shape[0] / 2.0 * tf.reduce_sum(tf.digamma(self.prec_a) - tf.log(self.prec_b))
- prop * tf.reduce_sum(plogvar) / 2.0
)
评论列表
文章目录