code: def is_injective(tag):
      """Check if a tag is injective
  
      Parameters
      ----------
      tag : str
          The input tag
  
      Returns
      -------
      ret : bool
          Whether a tag is injective
      """
      if tag in (ELEMWISE, BROADCAST, INJECTIVE):
          return True
      return (
          tag.startswith(ELEMWISE)
          or tag.startswith(BROADCAST)
          or tag.startswith(INJECTIVE)
      )
  
label_desc: Use image node_id 2 for calling a global function with example usage: is_injective(tag) and returns: unknown, True