code: def create_initial_beam_state(config):
      """Creates an instance of `BeamState` that can be used on the first
      call to `beam_step`.
  
      Args:
        config: A BeamSearchConfig
  
      Returns:
        An instance of `BeamState`.
      """
      return BeamSearchState(
          log_probs=tf.zeros([config.beam_width]),
          finished=tf.zeros([config.beam_width], dtype=tf.bool),
          lengths=tf.zeros([config.beam_width], dtype=tf.int32),
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from collections import namedtuple
  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: create_initial_beam_state(config) and returns: BeamSearchState