code: def _is_axis_required_in_distance_fn(key):
      return key in (
          configs.DistanceType.COSINE,
          configs.DistanceType.JENSEN_SHANNON_DIVERGENCE,
          configs.DistanceType.KL_DIVERGENCE,
      )
  
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 8 for calling a global function with example usage: _is_axis_required_in_distance_fn(key) and returns: unknown