code: def get_rng_state(self, device_index=None):
      if device_index is None:
          return torch.cuda.get_rng_state()
  
      return torch.cuda.get_rng_state(device_index)
  
imports: from deepspeed.accelerator.abstract_accelerator import DeepSpeedAccelerator
  import torch.cuda
label_desc: Use image node_id 11 for calling the CUDA_Accelerator obj's underlying member method code with example usage: obj.get_rng_state(device_index) and returns: torch, torch