(self,latent_dims)
| 52 | return z |
| 53 | class Decoder(nn.Module): |
| 54 | def __init__(self,latent_dims): |
| 55 | super(Decoder,self).__init__() |
| 56 | self.linear1 = nn.Linear(latent_dims,64) |
| 57 | self.linear2 = nn.Linear(64,128) |
| 58 | self.linear3 = nn.Linear(128,256) |
| 59 | |
| 60 | def forward(self,z): |
| 61 | z = F.relu(self.linear1(z)) |