code: def atom_features(atom):
      return torch.Tensor(
          list(onek_encoding_unk(atom.GetSymbol(), ELEM_LIST))
          + list(
              onek_encoding_unk(atom.GetDegree(), [0, 1, 2, 3, 4, 5])
          )
          + list(
              onek_encoding_unk(
                  atom.GetFormalCharge(), [-1, -2, 1, 2, 0]
              )
          )
          + list(
              onek_encoding_unk(int(atom.GetChiralTag()), [0, 1, 2, 3])
          )
          + [atom.GetIsAromatic()]
      )
  
file_path: DIG/dig/ggraph/method/JTVAE/fast_jtnn/mpn.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  import rdkit.Chem
  from .nnutils import *
  from .chemutils import get_mol
label_desc: Use image node_id 2 for calling a global function with example usage: atom_features(atom) and returns: torch