def forward(self, x):
x = x.view(-1, 784)
h1 = self.relu(self.fc1(x))
return self.fc21(h1), torch.exp(self.fc22(h1))
# VAE Decoder network
微信
公众号
扫码关注公众号