| 64 | return z |
| 65 | |
| 66 | class VariationalAutoencoder(nn.Module): |
| 67 | def __init__(self, latent_dims): |
| 68 | super(VariationalAutoencoder,self).__init__() |
| 69 | self.encoder = VariationalEncoder(latent_dims) |
| 70 | self.decoder = Decoder(latent_dims) |
| 71 | |
| 72 | def forward(self,x): |
| 73 | z = self.encoder(x) |
| 74 | return self.decoder(z) |