adgm.py 文件源码

def decode_xyz_a(self, x, y, z, test=False):
        x = self.to_variable(x)
        y = self.to_variable(y)
        z = self.to_variable(z)
        mean, ln_var = self.p_a_xyz(x, y, z, test=test)
        return F.gaussian(mean, ln_var)