Args: z: latent code tensor (B, C, H, W)
(self, z: torch.Tensor)
| 509 | |
| 510 | @torch.no_grad() |
| 511 | def decode(self, z: torch.Tensor): |
| 512 | """ |
| 513 | Args: |
| 514 | z: latent code tensor (B, C, H, W) |
| 515 | """ |
| 516 | z = z / self.scale + self.shift |
| 517 | z = self.post_quant_conv(z) |
| 518 | dec = self.decoder(z) |
| 519 | return dec |
| 520 | |
| 521 | def forward(self, input, sample_posterior=True): |
| 522 | posterior = self.encode(input, return_posterior=True) |