code: def _log_softmax(self, x):
      assert len(x.shape) == 2
      m = x.max(1)[:, np.newaxis]
      u = x - m
      return u - np.log(np.sum(np.exp(u), 1, keepdims=True))
  
imports: from absl.testing import parameterized
  import numpy
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.framework import weak_tensor
  from tensorflow.python.ops import gradient_checker_v2
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_impl
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import weak_tensor_ops
  from tensorflow.python.ops import weak_tensor_test_util
  import tensorflow.python.ops.nn_grad
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the LogSoftmaxTest obj's underlying member method code with example usage: obj._log_softmax(x) and returns: unknown