(self,x)
| 16 | self.kl = 0 |
| 17 | |
| 18 | def forward(self,x): |
| 19 | x = F.relu(self.linear1(x)) |
| 20 | x = F.relu(self.linear2(x)) |
| 21 | mu = self.linear3(x) |
| 22 | sigma = torch.exp(self.linear4(x)) |
| 23 | z = mu + sigma*self.N.sample(mu.shape) |
| 24 | self.kl = (sigma**2 + mu**2 - torch.log(sigma) - 1/2).sum() |
| 25 | return z |
| 26 | class Decoder(nn.Module): |
| 27 | def __init__(self,latent_dims): |
| 28 | super(Decoder,self).__init__() |
nothing calls this directly
no outgoing calls
no test coverage detected