class_docstr: Expands dims along a particular axis.
  
  Args:
      axis: Axis to expand.  A new dim is added before this axis. May be a
        negative value.  Must not be a tensor.
  
  Input shape:
      `(batch_size,) + shape`
  
  Output shape:
      `(batch_size,) + shape + [1]`, if `axis == -1`.
  
      `(batch_size,) + shape[:axis + 1] + [1] + shape[axis + 1:]`,
      if `axis < -1`.
  
      `(batch_size,) + shape[:axis] + [1] + shape[axis:]`, if `axis >= 0`.
code: def get_config(self):
      config = {"axis": self.axis}
      base_config = super(ExpandDims, self).get_config()
      return dict(list(base_config.items()) + list(config.items()))
  
file_path: agents/tf_agents/networks/expand_dims_layer.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
label_desc: Use image node_id 4 for calling the ExpandDims obj's underlying member method code with example usage: obj.get_config() and returns: dict