1 coral_logits_to_prob
rri_metrics.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. |