adgm.py 文件源码

def sample_x_label(self, x, argmax=True, test=False):
        x = self.to_variable(x)
        mean, ln_var = self.q_a_x(x, test=test)
        a = F.gaussian(mean, ln_var)
        return self.sample_ax_label(a, x, argmax=argmax, test=test)