model.py 文件源码

def encode_a(self, x):
        a_params = self.qlina0(x)
        a_params = self.qlina_batch_norm_0(a_params)
        a_params = F.crelu(a_params)

        for i in range(self.num_layers-1):
            layer_name = 'qlina' + str(i+1)
            a_params = self[layer_name](a_params)
            layer_name = 'qlina_batch_norm_' + str(i+1)
            a_params = self[layer_name](a_params)
            a_params = F.crelu(a_params)

        self.qmu_a = self.qlina_mu(a_params)
        self.qln_var_a = self.qlina_ln_var(a_params)

        return self.qmu_a, self.qln_var_a