added NaN test for log_softmax
added NaN test for softmax
added NaN test for log_sum_exp