code: def __init__(
      self,
      num_embeddings: int,
      cuda_row_num: int,
      assigned_rank: int = 0,
      buffer_size=50_000,
      ids_freq_mapping=None,
      initial_weight: torch.tensor = None,
      name: str = "",
  ):
      self.num_embeddings = num_embeddings
      self.cuda_row_num = cuda_row_num
      self.assigned_rank = assigned_rank
      self.buffer_size = buffer_size
      self.ids_freq_mapping = ids_freq_mapping
      self.initial_weight = initial_weight
      self.name = name
  
label_desc: Use image node_id 1 to create a new TablewiseEmbeddingBagConfig object  with example: obj = TablewiseEmbeddingBagConfig(num_embeddings, cuda_row_num, assigned_rank, buffer_size, ids_freq_mapping, initial_weight, name)