I try to apply softmax on output it returns probability for the single record only instead of returning the probability for whole test data,
it returns probability correctly for the single record torch.softmax(output[0], dim=0).
but this one not working torch.softmax(output, dim=1) its returning some other value instead of probability
@ptrblck