code: def normalize_embeds(embeds):
      return embeds * (
          tf.cast(tf.shape(embeds)[-1], dtype=tf.float32) ** -0.5
      )
  
imports: import functools
  import numpy
  import tensorflow
  from finetune.base import BaseModel
  from finetune.encoding.target_encoders import Seq2SeqLabelEncoder
  from finetune.input_pipeline import BasePipeline
  from finetune.model import PredictMode
  from finetune.util.beam_search import beam_search, get_state_shape_invariants
  from tensorflow.python.util import nest
  import itertools
  import sys
label_desc: Use image node_id 1 for calling a global function with example usage: normalize_embeds(embeds) and returns: unknown