adgm.py 文件源码

def decode_yz_a(self, y, z, test=False):
        y = self.to_variable(y)
        z = self.to_variable(z)
        mean, ln_var = self.p_a_yz(y, z, test=test)
        return F.gaussian(mean, ln_var)