code: def merge_beam_dim(tensor):
      """Reshapes first two dimensions in to single dimension."""
      shape = shape_list(tensor)
      shape[0] *= shape[1]  # batch -> batch * beam_size
      shape.pop(1)  # Remove beam dim
      return tf.reshape(tensor, 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 3 for calling a global function with example usage: merge_beam_dim(tensor) and returns: tf