Hello,
I have a question about Softmax() and CrossEntropyLoss().
In a multi-classification task, I set dim=1 in Softmax(). I wanna know if I need to set the similar parameter in CrossEntropyLoss(). However, I did not find the similar parameter as dim in softmax().
Thanks!