code: def get_heatmap_values(heatmap, indices):
      b, _, _, n = heatmap.shape
      values = tf.reshape(heatmap, [b, -1, n])
      values = tf.gather(values, indices, batch_dims=1)
      return values
  
file_path: graphics/tensorflow_graphics/projects/points_to_3Dobjects/models/centernet_utils.py
imports: import tensorflow
label_desc: Use image node_id 2 for calling a global function with example usage: get_heatmap_values(heatmap, indices) and returns: values