class_docstr: Data structure that contains information about a atom and its contracted
  gaussian type orbital (CGTO).
  
  Examples
  --------
  >>> import torch
  >>> from deepchem.utils.dft_utils import AtomCGTOBasis, CGTOBasis
  >>> alphas = torch.ones(1)
  >>> coeffs = torch.ones(1)
  >>> cgto = CGTOBasis(angmom=0, alphas=alphas, coeffs=coeffs)
  >>> atomcgto = AtomCGTOBasis(atomz=1, bases=[cgto], pos=[[0.0, 0.0, 0.0]])
  >>> atomcgto
  AtomCGTOBasis(atomz=1, bases=[CGTOBasis(angmom=0, alphas=tensor([1.]), coeffs=tensor([1.]), normalized=False)], pos=tensor([[0., 0., 0.]]))
docstr: Initialize the AtomCGTOBasis object.
  
  Parameters
  ----------
  atomz: ZType
      Atomic number of the atom.
  bases: List[CGTOBasis]
      List of CGTOBasis objects.
  pos: AtomPosType
      Position of the atom. Shape: (ndim,)
docstr_tok: ["Initialize","the","AtomCGTOBasis","object",".","Parameters","--","--","--","--","--","atomz",":","ZType","Atomic","number","of","the","atom",".","bases",":","List","[","CGTOBasis","]","List","of","CGTOBasis","objects",".","pos",":","AtomPosType","Position","of","the","atom",".","Shape",":","(","ndim",",",")"]
code: def __init__(
      self, atomz: ZType, bases: List[CGTOBasis], pos: AtomPosType
  ):
      """Initialize the AtomCGTOBasis object.
  
      Parameters
      ----------
      atomz: ZType
          Atomic number of the atom.
      bases: List[CGTOBasis]
          List of CGTOBasis objects.
      pos: AtomPosType
          Position of the atom. Shape: (ndim,)
  
      """
      self.atomz = atomz
      self.bases = bases
      self.pos = torch.tensor(pos)
  
imports: from typing import Union, TypeVar, Generic, Optional, Callable, List, Dict
  from dataclasses import dataclass
  import torch
  import numpy
  from deepchem.utils import gaussian_integral
label_desc: Use image node_id 1 to create a new AtomCGTOBasis object  with example: obj = AtomCGTOBasis(atomz, bases, pos)