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