code: def to_device(self, device, non_blocking=True):
          
          return Trajectories(
              num_steps=self.num_steps,
              num_envs=self.num_envs,
              environment_information=self.environment_information,
              transition_tensors={k: v.to(device, non_blocking=non_blocking) for k, v in self.transition_tensors.items()},
              rollout_tensors={k: v.to(device, non_blocking=non_blocking) for k, v in self.rollout_tensors.items()},
              extra_data=self.extra_data
          )
docs: Move a rollout to a selected device