def to_device(self, device, non_blocking=True):
return Transitions(<EOL>size=self.size,<EOL>environment_information=self.environment_information,<EOL>transition_tensors={k: v.to(device, non_blocking=non_blocking) for k, v in self.transition_tensors.items()},<EOL>extra_data=self.extra_data<EOL>)<EOL>
Move a rollout to a selected device