Hello,
I am working multi-class. When I study at Keras I can use “predict_proba” function for can see probability of  every class.
Model is Sequential() and I used CNN.
code: y_score = model.predict_proba(testX)
I want to learn, is there a function at Pytorch like “predict_proba” .
Thank you for helping