code: def get_memory(a: torch.Tensor) -> int:
      """Returns the size of the tensor in bytes.
  
      Examples
      --------
      >>> import torch
      >>> from deepchem.utils.pytorch_utils import get_memory
      >>> a = torch.randn(100, 100, dtype=torch.float64)
      >>> get_memory(a)
      80000
  
      Parameters
      ----------
      a: torch.Tensor
          The tensor to be measured.
  
      Returns
      -------
      size: int
          The size of the tensor in bytes.
  
      """
      size = a.numel() * a.element_size()
      return size
  
file_path: deepchem/deepchem/utils/pytorch_utils.py
imports: import scipy
  import torch
  from typing import Callable, Union, List, Generator, Tuple
label_desc: Use image node_id 5 for calling a global function with example usage: get_memory(a) and returns: size