code: def unmerge_beam_dim(tensor, batch_size, beam_size):
      """Reshapes first dimension back to [batch_size, beam_size]."""
      shape = shape_list(tensor)
      new_shape = [batch_size] + [beam_size] + shape[1:]
      return tf.reshape(tensor, new_shape)
  
file_path: delta/delta/layers/utils_tf.py
imports: import math
  import numpy
  import tensorflow
  from tensorflow.python.util import nest
label_desc: Use image node_id 4 for calling a global function with example usage: unmerge_beam_dim(tensor, batch_size, beam_size) and returns: tf