[FEATURE] Is there any way to save the logits of the classification task?

[yufu]

Please use this template to submit your need.

---

Is your feature request related to a problem? Please describe. A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]

Describe the solution you'd like A clear and concise description of what you want to happen.

Describe alternatives you've considered A clear and concise description of any alternative solutions or features you've considered.

Additional context Add any other context or screenshots about the feature request here.

[FateScript]

Please use template to specify your problem.