code: def __init__(self, emb_dim):
      super(AtomEncoder, self).__init__()
  
      self.atom_embedding_list = nn.LayerList()
  
      for i, dim in enumerate(full_atom_feature_dims):
          weight_attr = nn.initializer.XavierUniform()
          emb = paddle.nn.Embedding(
              dim, emb_dim, weight_attr=weight_attr
          )
          self.atom_embedding_list.append(emb)
  
file_path: PGL/ogb_examples/graphproppred/ogbg_molpcba/mol_encoder.py
imports: import paddle
  import paddle.nn
  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: nn with example: obj = AtomEncoder(emb_dim)