code: def after_update(self):
      self.observations[0].copy_(self.observations[-1])
      self.states[0].copy_(self.states[-1])
      self.masks[0].copy_(self.masks[-1])
  
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 4 for calling the RolloutStorage obj's underlying member method code with example usage: obj.after_update() without return types