code: def _strip_batch_dim(type_):
      return instructions.pattern_map(
          lambda t: instructions.TensorType(t.dtype, t.shape[1:]),
          type_,
          leaf_type=instructions.TensorType,
      )
  
file_path: probability/tensorflow_probability/python/experimental/auto_batching/type_inference.py
imports: import contextlib
  import functools
  from absl import logging
  import six
  from tensorflow_probability.python.experimental.auto_batching import instructions
label_desc: Use image node_id 14 for calling a global function with example usage: _strip_batch_dim(type_) and returns: instructions