tf.compat.v1.math.log_softmax