vae_m1.py 文件源码

def gaussian_nll_keepbatch(self, x, mean, ln_var):
        x_prec = F.exp(-ln_var)
        x_diff = x - mean
        x_power = x_diff ** 2 * x_prec * 0.5
        return F.sum((math.log(2.0 * math.pi) + ln_var) * 0.5 + x_power, axis=1)