Searched refs:logsumexp_logit (Results 1 – 1 of 1) sorted by relevance
87 def logsumexp_logit(embeddings): function97 weights, sampled, partition_strategy, transform_fn=logsumexp_logit)