def logpdf_conditional(self, w, z): mean = self.conditional_mean(z) var = self.conditional_variance(z) return norm.logpdf(w, loc=mean, scale=np.sqrt(var))