def _log_prob(self, given):
temperature, logits = self.path_param(self.temperature), \
self.path_param(self.logits)
log_given = tf.log(given)
log_1_minus_given = tf.log(1 - given)
log_temperature = tf.log(temperature)
if self._check_numerics:
log_given = tf.check_numerics(log_given, "log(given)")
log_1_minus_given = tf.check_numerics(
log_1_minus_given, "log(1 - given)")
log_temperature = tf.check_numerics(
log_temperature, "log(temperature)")
logistic_given = log_given - log_1_minus_given
temp = temperature * logistic_given - logits
return log_temperature - log_given - log_1_minus_given + \
temp - 2 * tf.nn.softplus(temp)
评论列表
文章目录