(self, z, logdet=0.0)
| 199 | return self.encode(input, logdet) |
| 200 | |
| 201 | def encode(self, z, logdet=0.0): |
| 202 | for layer, shape in zip(self.layers, self.output_shapes): |
| 203 | z, logdet = layer(z, logdet, reverse=False) |
| 204 | return z, logdet |
| 205 | |
| 206 | def decode(self, z, temperature=None): |
| 207 | for layer in reversed(self.layers): |