model.py 文件源码

def decode(self, z):
        h = self.plin0(z)
        h = self.plin_batch_norm_0(h)
        h = F.crelu(h)

        for i in range(self.num_layers-1):
            layer_name = 'plin' + str(i+1)
            h = self[layer_name](h)

            layer_name = 'plin_batch_norm_' + str(i+1)
            h = self[layer_name](h)
            h = F.crelu(h)        

        self.p_ber_prob_logit = self.plin_ber_prob(h)

        return self.p_ber_prob_logit