def forward(self, x):
a = self.mlp(x)
return a[:, 0:self.z_size], softplus(a[:, self.z_size:])
# Takes a latent code, z_what, to pixel intensities.
微信
公众号
扫码关注公众号