code: def memory_stats_as_nested_dict(
      device: Union[Device, int] = None
  ) -> Dict[str, Any]:
      r"""Return the result of :func:`~torch.cuda.memory_stats` as a nested dictionary."""
      if not is_initialized():
          return {}
      device = _get_device_index(device, optional=True)
      return torch._C._cuda_memoryStats(device)
  
imports: import collections
  import contextlib
  import ctypes
  import pickle
  import sys
  import warnings
  from inspect import signature
  from typing import Any, Dict, Optional, Tuple, Union
  import torch
  from torch import _C
  from torch.types import Device
  from .None import _get_device_index, _get_nvml_device_index, _lazy_init, is_initialized
  from ._memory_viz import memory, segments
  from ._utils import _dummy_type
label_desc: Use image node_id 8 for calling a global function with example usage: memory_stats_as_nested_dict(device) and returns: torch, dict