def _log_prob(self, counts): counts = self._maybe_assert_valid_sample(counts) ordered_prob = ( tf.lbeta(self.concentration + counts) - tf.lbeta(self.concentration)) return ordered_prob + distribution_util.log_combinations( self.total_count, counts)
def _log_normalization(self, counts): counts = self._maybe_assert_valid_sample(counts) return -distribution_util.log_combinations(self.total_count, counts)