mMagmer:

eta = nn.functional.softmax(logits,-1)


it should work.
replace
for imageName,realLabel,actualLabel,logitValue in zip(tested_file_name,real_label,actual_pred,logits):

with
for imageName,realLabel,actualLabel,logitValue in zip(tested_file_name,real_label,actual_pred,eta):

some where in your for loop print it.