hi @Amitabh_G
You are suppose to use tf.nn.log_softmwx that is correct
The issue is not with that. you are probably not passing the right argument somewhere in your codes, so either share a complete screenshot of your error or
refer the below link to find the solution.
Also remember to provide a complete screenshot of the error,.if it is lengthy take two separate screenshots as that gives information on why you got the error.
please make sure not to post codes here.
Regards
DP