vae_m2.py 文件源码

def __call__(self, z, y, test=False, apply_f=False):
        mean, ln_var = self.forward_one_step(z, y, test=test, apply_f=False)
        if apply_f:
            return F.gaussian(mean, ln_var)
        return mean, ln_var