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)