code: def normalize_embeddings(embeddings: torch.Tensor):
      # assume [batch, embed_dim] dimensions
      # eps to make sure everything works in fp16
      return torch.nn.functional.normalize(embeddings, eps=1e-6)
  
label_desc: Use image node_id 1 for calling a global function with example usage: normalize_embeddings(embeddings) and returns: torch