code: def cuda(self, device=None):
      self.observations = self.observations.cuda(device=device)
      self.states = self.states.cuda(device=device)
      self.rewards = self.rewards.cuda(device=device)
      self.value_preds = self.value_preds.cuda(device=device)
      self.returns = self.returns.cuda(device=device)
      self.action_log_probs = self.action_log_probs.cuda(device=device)
      self.actions = self.actions.cuda(device=device)
      self.masks = self.masks.cuda(device=device)
  
file_path: SerpentAI/serpent/machine_learning/reinforcement_learning/ppo/rollout_storage.py
imports: import torch
  from torch.utils.data.sampler import BatchSampler, SubsetRandomSampler
label_desc: Use image node_id 2 for calling the RolloutStorage obj's underlying member method code with example usage: obj.cuda(device) without return types