(self, device=None, non_blocking=False, memory_format: torch.memory_format = torch.preserve_format)
| 385 | return self.to(device=torch.device("cpu"), memory_format=memory_format) |
| 386 | |
| 387 | def cuda(self, device=None, non_blocking=False, memory_format: torch.memory_format = torch.preserve_format): |
| 388 | device = torch.device(device or "cuda") |
| 389 | return self.to(device=device, non_blocking=non_blocking, memory_format=memory_format) |
| 390 | |
| 391 | def clone(self) -> "LazyTensor": |
| 392 | def factory_fn(t: torch.Tensor, **kw): |