Ejemplo n.º 1
0
 def random_sample(self, num_samples=8):
     z = Normal(torch.tensor(0.), torch.tensor(1.)).sample(
         (num_samples, self.z_dim))
     if self.use_cuda:
         z = z.cuda()
     mu = self.decoder(z)
     img = Normal(mu, self.noise_level).sample().cpu()
     return img