 这里logits是不是应该为self.probabilities