class_docstr: The complete Atom Encoder used in OGB dataset
  
  Parameters:
  num_classes: Not used!
class_docstr_tok: ["The","complete","Atom","Encoder","used","in","OGB","dataset","Parameters",":","num_classes",":","Not","used","!"]
code: def forward(self, batch):
      encoded_features = 0
      for i in range(batch.x.shape[1]):
          encoded_features += self.atom_embedding_list[i](batch.x[:, i])
  
      batch.x = encoded_features
      return batch
  
file_path: GraphGym/graphgym/models/feature_encoder_pyg.py
imports: import torch
  from ogb.utils.features import get_atom_feature_dims, get_bond_feature_dims
  import graphgym.register
label_desc: Use image node_id 2 for calling the AtomEncoder obj's underlying member method code with example usage: obj.forward(batch) and returns: batch