(self)
| 748 | return self |
| 749 | |
| 750 | def cpu(self): |
| 751 | return self.to(device="cpu") |
| 752 | |
| 753 | def cuda(self, device: Optional[int | device | str] = None, non_blocking: bool = False): |
| 754 | return self.to(device="cuda" if device is None else device, non_blocking=non_blocking) |