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
评论列表
文章目录