adgm.py 文件源码

def encode_x_z(self, x, test=False, argmax_y=True):
        x = self.to_variable(x)
        mean, ln_var = self.q_a_x(x, test=test)
        a = F.gaussian(mean, ln_var)
        y = self.sample_x_y(x, argmax=argmax_y, test=test)
        mean, ln_var = self.q_z_axy(a, x, y, test=test)
        return F.gaussian(mean, ln_var)