def _log_prob(self, x): return (-0.5 * math.log(2. * math.pi) - math_ops.log(self.sigma) -0.5 * math_ops.square(self._z(x)))