logits assumed some form of softmax or logistic, which may not be the case