vbutils.py 文件源码

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
               )