def decode(self, z):
h = F.crelu(self.plin0(z))
for i in range(self.num_layers-1):
layer_name = 'plin' + str(i+1)
h = F.crelu(self[layer_name](h))
self.p_ber_prob_logit = self.plin_ber_prob(h)