code: def _log_softmax(
      logits, axis=None, name=None
  ):  # pylint: disable=unused-argument
      logits = _convert_to_tensor(logits)
      axis = -1 if axis is None else axis
      y = logits - _max_mask_non_finite(
          logits, axis=axis, keepdims=True
      )
      return y - np.log(
          np.sum(np.exp(y), axis=_astuple(axis), keepdims=True)
      )
  
imports: import collections
  import functools
  import numpy
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy.numpy_array import _reverse
  from tensorflow_probability.python.internal.backend.numpy.numpy_array import one_hot
  from tensorflow_probability.python.internal.backend.numpy.ops import _convert_to_tensor
label_desc: Use image node_id 18 for calling a global function with example usage: _log_softmax(logits, axis, name) and returns: unknown