vaewgan.py 文件源码

def __call__(self, x, train=True):
        h = F.leaky_relu(self.c0_0(x))
        h = F.leaky_relu(self.bn0_1(self.c0_1(h), test=not train))
        h = F.leaky_relu(self.bn1_1(self.c1_1(h), test=not train))
        h = F.leaky_relu(self.bn2_1(self.c2_1(h), test=not train))
        h = F.leaky_relu(self.bn3_0(self.c3_0(h), test=not train))

        mean = self.mean(h)
        ln_var = self.ln_var(h)

        return mean, ln_var