Update main.py (pytorch#520)

davidyang19971209 · Mar 6, 2019 · 67e5b06 · 67e5b06
1 parent 87d9a1e
commit 67e5b06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vae/main.py b/vae/main.py
@@ -53,7 +53,7 @@ def encode(self, x):
     def reparameterize(self, mu, logvar):
         std = torch.exp(0.5*logvar)
         eps = torch.randn_like(std)
-        return eps.mul(std).add_(mu)
+        return mu + eps*std
 
     def decode(self, z):
         h3 = F.relu(self.fc3(z))