model.py 文件源码

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

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

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

       return self.qmu, self.qln_var