Soft max is output greated than 1

HI team , i am using AutoModelForSequenceClassification , model is roberta , using it for text classification.

there are 3 classes

output is [-3.7550,-4.4172,7.8079]

i need to convert this to probabilities
should I apply soft max to this to get the probabilities , if i do that i am getting outputs greater than one


Try specifying the axis you are applying the softmax function on. Like this:

torch.nn.functional.softmax(torch.tensor([[-3.7550, -4.4172, 7.8079]]), dim=1)