def normalize(self, logits: torch.Tensor) -> torch.Tensor:
    '\n        Given logits regarding the entire vocabulary, return the probs over the label words set.\n\n        Args:\n            logits (:obj:`Tensor`): The logits over the entire vocabulary.\n\n        Returns:\n            :obj:`Tensor`: The logits over the label words set.\n\n        '
    batch_size = logits.shape[0]
    return F.softmax(logits.reshape(batch_size, (- 1)), dim=(- 1)).reshape(*logits.shape)
Given logits regarding the entire vocabulary, return the probs over the label words set.

Args:
    logits (:obj:`Tensor`): The logits over the entire vocabulary.

Returns:
    :obj:`Tensor`: The logits over the label words set.
def normalize(self, logits: torch.Tensor) -> torch.Tensor:
    '\n        Given logits regarding the entire vocabulary, return the probs over the label words set.\n\n        Args:\n            logits (:obj:`Tensor`): The logits over the entire vocabulary.\n\n        Returns:\n            :obj:`Tensor`: The logits over the label words set.\n\n        '
    batch_size = logits.shape[0]
    return F.softmax(logits.reshape(batch_size, (- 1)), dim=(- 1)).reshape(*logits.shape)