code: def _expand_right(a, n, pos):
      """Insert multiple dimensions of size 1 at position `pos` in tensor's shape.
  
      Equivalent to performing `expand_dims(..., pos)` `n` times.
  
      Args:
        a: tensor into which extra dimensions will be inserted.
        n: number of inserted dimensions.
        pos: choice of dimension for insertion. Must be negative.
  
      Returns:
        Tensor with inserted dimensions.
      """
  
      axis = ps.rank(a) + pos + 1
      return tf.reshape(
          a,
          ps.concat(
              [
                  ps.shape(a)[:axis],
                  ps.ones([n], dtype=tf.int32),
                  ps.shape(a)[axis:],
              ],
              axis=0,
          ),
      )
  
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import bernoulli
  from tensorflow_probability.python.distributions import categorical
  from tensorflow_probability.python.distributions import joint_distribution
  from tensorflow_probability.python.distributions import joint_distribution_coroutine
  from tensorflow_probability.python.distributions import sample
  from tensorflow_probability.python.experimental.marginalize.logeinsumexp import logeinsumexp
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import samplers
label_desc: Use image node_id 5 for calling a global function with example usage: _expand_right(a, n, pos) and returns: tf