code: def __init__(self, labels: List):
      # save labels into a dict having label as key and a tensor of size
      #  len(labels) as value. The tensor contains ones up to the index of
      #  the label and zeros after.
      self.label_names = labels
      self.labels = [
          torch.nn.functional.one_hot(
              torch.tensor([i]), len(labels)
          ).reshape(-1)
          for i in range(len(labels))
      ]
  
imports: from typing import List
  import torch
label_desc: Use image node_id 1 to create a new LabelsInjector object  with example: obj = LabelsInjector(labels)