code: def compute_batch_indices(batch_size, beam_size):
      """Computes the i'th coordinate that contains the batch index for gathers."""
      batch_pos = tf.range(batch_size * beam_size) // beam_size
      batch_pos = tf.reshape(batch_pos, [batch_size, beam_size])
      return batch_pos
  
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 7 for calling a global function with example usage: compute_batch_indices(batch_size, beam_size) and returns: batch_pos