code: def __init__(self, emb_dim, optional_full_atom_features_dims=None):
      super(AtomEncoder, self).__init__()
  
      self.atom_embedding_list = torch.nn.ModuleList()
  
      if optional_full_atom_features_dims is not None:
          full_atom_feature_dims = optional_full_atom_features_dims
      else:
          full_atom_feature_dims = get_atom_feature_dims()
  
      for i, dim in enumerate(full_atom_feature_dims):
          emb = torch.nn.Embedding(dim, emb_dim)
          torch.nn.init.xavier_uniform_(emb.weight.data)
          self.atom_embedding_list.append(emb)
  
file_path: ogb/ogb/graphproppred/mol_encoder.py
imports: import torch
  from ogb.utils.features import get_atom_feature_dims, get_bond_feature_dims
label_desc: Use image node_id 1 to create a new AtomEncoder object from inherited base classes: torch.nn with example: obj = AtomEncoder(emb_dim, optional_full_atom_features_dims)