model.py 文件源码

def encode(self, x):
        h = self.qlin0(x)
        h = self.qlin_batch_norm_0(h)
        h = F.crelu(h)

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

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

        self.qmu = self.qlin_mu(h)
        self.qln_var = self.qlin_ln_var(h)