I started to learn about pytorch lately after using tensorflow for almost 1 year, i am confused about something:
In Tensorflow when we have multiclassification problem we set at the last activation layer the number of classes and the type of activation function which is "Softmax" and using “Cross-entropy loss”
so in Pytorch when building a network we set last layer to nn.linear().
can anyone clarify the concept?
Thanks a lot.