feat: add a use_softmax
field in variant class
#533
Merged
use_softmax
field in variant class
#533