def _calc_logits(self, X):
return self.forward(X)<EOL>
Calculate the logits.

:param x: The input (batch) of the model.
:type x: torch.Tensor of shape (batch_size, num_classes)
:return: The output logits of model.
:rtype: torch.Tensor of shape (batch_size, num_classes)