code: def forward(self, x):
      x_embedding = 0
      for i in range(x.shape[1]):
          x_embedding += self.atom_embedding_list[i](x[:, i])
  
      return x_embedding
  
file_path: PGL/ogb_examples/lsc/PCQM4M/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 2 for calling the AtomEncoder obj's underlying member method code with example usage: obj.forward(x) and returns: x_embedding