def decode_a(self, z):
# net_input = F.concat((x,z), axis=1)
h = self.plina0(z)
h = self.plina_batch_norm_0(h)
h = F.crelu(h)
for i in range(self.num_layers-1):
layer_name = 'plina' + str(i+1)
h = self[layer_name](h)
layer_name = 'plina_batch_norm_' + str(i+1)
h = self[layer_name](h)
h = F.crelu(h)
self.pmu_a = self.plina_mu(h)
self.pln_var_a = self.plina_ln_var(h)
return self.pmu_a, self.pln_var_a
评论列表
文章目录