model.py 文件源码

def decode(self,z):
        h = self.plinx0(z)
        h = self.plinx_batch_norm_0(h)
        h = F.crelu(h)

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

        self.p_ber_prob_logit = self.plinx_ber_prob(h)

        return self.p_ber_prob_logit