Compute disagreement among ensemble members.
Disagreement is measured as the variance of predictions.
- Parameters:
predictions (Tensor) – Tensor of shape (num_models, batch_size, num_classes)
- Return type:
Tensor
- Returns:
Disagreement score of shape (batch_size,)