def _log_prob(self, given): log_p = tf.log(self._prob(given)) if self._check_numerics: log_p = tf.check_numerics(log_p, "log_p") return log_p