adgm.py 文件源码

def encode_x_y_distribution(self, x, test=False, softmax=True):
        x = self.to_variable(x)
        mean, ln_var = self.q_a_x(x, test=test)
        a = F.gaussian(mean, ln_var)
        y = self.q_y_ax(a, x, test=test)
        if softmax:
            return F.softmax(y)
        return y