class_docstr: Only encode the first dimension of atom integer features.
  This feature encodes just the atom type
  Args:
      emb_dim (int): Output embedding dimension
      num_classes: None
class_docstr_tok: ["Only","encode","the","first","dimension","of","atom","integer","features",".","This","feature","encodes","just","the","atom","type","Args",":","emb_dim","(","int",")",":","Output","embedding","dimension","num_classes",":","None"]
code: def forward(self, batch):
      batch.node_feature = self.atom_type_embedding(
          batch.node_feature[:, 0]
      )
  
      return batch
  
file_path: GraphGym/graphgym/models/feature_encoder.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 SingleAtomEncoder obj's underlying member method code with example usage: obj.forward(batch) and returns: batch