vae_m1.py 文件源码

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

# Network structure is same as the Encoder