Method decode

tutorials/03-advanced/variational_autoencoder/main.py:57–59

(self, z)

        return mu + eps * std

    def decode(self, z):
        h = F.relu(self.fc4(z))
        return F.sigmoid(self.fc5(h))
    
    def forward(self, x):
        mu, log_var = self.encode(x)