Convert all tensors to CPU in data.
(self)
| 490 | |
| 491 | # Tensor-like methods |
| 492 | def cpu(self) -> 'BaseDataElement': |
| 493 | """Convert all tensors to CPU in data.""" |
| 494 | new_data = self.new() |
| 495 | for k, v in self.items(): |
| 496 | if isinstance(v, (torch.Tensor, BaseDataElement)): |
| 497 | v = v.cpu() |
| 498 | data = {k: v} |
| 499 | new_data.set_data(data) |
| 500 | return new_data |
| 501 | |
| 502 | # Tensor-like methods |
| 503 | def cuda(self) -> 'BaseDataElement': |