model.py 文件源码

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

        for i in range(self.num_layers-1):
            layer_name = 'qlina' + str(i+1)
            a_params = F.crelu(self[layer_name](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