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])
          + onek_encoding_unk(int(atom.GetChiralTag()), [0, 1, 2, 3])
          + [atom.GetIsAromatic()]
      )
  
imports: import dgl
  import dgl.function
  import rdkit.Chem
  import torch
  import torch.nn
  import torch.nn.functional
  from dgl import line_graph, mean_nodes
  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