As far as I know, CrossEntropy() equals to
torch.mean(torch.sum(-target * nn.LogSoftmax(input), dim=1))
How can I express nn.LogSoftmax in equation?
nn.LogSoftmax  equals to torch.mean(torch.sum(F.softmax(input)*torch.log(F.softmax(input)),1) ?