code: def get_atomz(element: Union[str, ZType]) -> ZType:
      """Returns the atomic number for the given element
  
      Examples
      --------
      >>> from deepchem.utils import get_atomz
      >>> element_symbol = "Al" # Aluminium
      >>> get_atomz(element_symbol)
      13
      >>> get_atomz(17)
      17
  
      Parameters
      ----------
      element: Union[str, ZType]
          String symbol of Element or Atomic Number.
          Ex: H, He, C
  
      Returns
      -------
      atom_n: ZType
          Atomic Number of the given Element.
  
      References
      ----------
      Kasim, Muhammad F., and Sam M. Vinko. "Learning the exchange-correlation
      functional from nature with fully differentiable density functional
      theory." Physical Review Letters 127.12 (2021): 126403.
      https://github.com/diffqc/dqc/blob/master/dqc/utils/periodictable.py
  
      """
      if isinstance(element, str):
          try:
              atom_z = periodic_table_atomz[element]
          except KeyError:
              raise KeyError(
                  "Element Does Not Exists or Not Documented: %s"
                  % element
              )
          return atom_z
      elif isinstance(element, torch.Tensor):  # Just return itself.
          try:
              assert element.numel() == 1
          except:
              raise AssertionError("Only 1 element Tensor Allowed.")
          return element
      else:  # float or int | Just return itself.
          return element
  
imports: from typing import Union
  import numpy
  import torch
label_desc: Use image node_id 1 for calling a global function with example usage: get_atomz(element) and returns: atom_z, element, element