Actually, I was regenerating the results of one paper, where they have used softmax at the last layer. They did not mention the loss function. So it is kind of mandatory to apply softmax at the last layer. Is there any alternative that does exactly as same as mentioned. or is there any custom implementation of cross-entropy loss (the basic cross-entropy loss  -y_i log y_i  where is i is true label.)