code: def log_softmax(c, axis=None):
      c = as_tensor_variable(c)
      return LogSoftmax(axis=axis)(c)
  
imports: from textwrap import dedent
  import numpy
  import scipy
  from pytensor.graph.basic import Apply
  from pytensor.graph.replace import _vectorize_node
  from pytensor.link.c.op import COp
  from pytensor.tensor.basic import as_tensor_variable
  from pytensor.tensor.elemwise import get_normalized_batch_axes
  from pytensor.tensor.math import gamma, gammaln, neg, sum
label_desc: Use image node_id 2 for calling a global function with example usage: log_softmax(c, axis) and returns: LogSoftmax