Jona:

You mean, take the predictions and see their probability distributions?


Yes, exactly. You could use a threshold of e.g. 0 to create the predictions and check the predicted class labels (similar to how you would calculate the accuracy). I assume torch.unique would return only the majority class.