adgm.py 文件源码

def encode_axy_z(self, a, x, y, test=False):
        a = self.to_variable(a)
        x = self.to_variable(x)
        y = self.to_variable(y)
        mean, ln_var = self.q_z_axy(a, x, y, test=test)
        return F.gaussian(mean, ln_var)