KFrank:

Let input be a tensor of shape (nBatch, nClass) .


The input is not shape of ‘(n batch, nClass)’. The input is in shape (‘nBatch, mLogit’). i.e., each row has 1000 elements, corresponding to each classes.
I want to get sum over all classes.