some question about the code

[Shu-HowTing]

class OrdinalLogisticModel(nn.Module):
"""
 Parameters
    ----------
    predictor : nn.Module
        When called, must return a torch.FloatTensor with shape [batch_size, 1]
"""

hello , thank u for sharing! I have a question here. why return a torch.FloatTensor with shape [batch_size, 1]. Shouldn't it be return a torch.FloatTensor with shape [batch_size, num_class - 1]?