vaegan.py 文件源码

def __call__(self, x, train=True):
        h1 = F.leaky_relu(self.enc1(x))
        h2 = F.leaky_relu(self.norm2(self.enc2(h1), test=not train))
        h3 = F.leaky_relu(self.norm3(self.enc3(h2), test=not train))
        h4 = F.leaky_relu(self.norm4(self.enc4(h3), test=not train))
        mean = self.mean(h4)
        ln_var = self.ln_var(h4)

        return mean, ln_var