1 coral_logits_to_prob

rri_metrics.coral.coral_logits_to_prob(logits)

Convert CORAL logits to a proper class distribution.

1.1 Parameters

Name Type Description Default
logits Tensor Tensor["... K-1"] required

1.2 Returns

Name Type Description
Tensor Tensor["... K"] probabilities that sum to 1 along the last dim.