code: def _apply_transform(batched_tensor, transform_type, axis=None):
      """Applies the given transform function to `batched_tensor` along `axis`."""
      if transform_type == configs.TransformType.SOFTMAX:
          return tf.nn.softmax(batched_tensor, axis=axis)
      else:
          raise ValueError("Invalid TransformType %s." % transform_type)
  
file_path: neural-structured-learning/neural_structured_learning/lib/distances.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import neural_structured_learning.configs
  import tensorflow
label_desc: Use image node_id 6 for calling a global function with example usage: _apply_transform(batched_tensor, transform_type, axis) and returns: tf