(self, mu, log_var)
| 568 | return out |
| 569 | |
| 570 | def reparameterize(self, mu, log_var): |
| 571 | std = torch.exp(0.5 * log_var) |
| 572 | eps = torch.randn_like(std) |
| 573 | return eps * std + mu |
| 574 | |
| 575 | def sample(self, imgs, deterministic=False): |
| 576 | mu, log_var = self.encode(imgs) |