The outputs would be the featurized data, you could simply apply a softmax layer to the output of a forward pass. Something like:
model = nn.Sequential(...)
probs = nn.Softmax(dim=1)

outputs = model(input)
probs(outputs)