Convert all tensors to np.ndarray in data.
(self)
| 555 | |
| 556 | # Tensor-like methods |
| 557 | def numpy(self) -> 'BaseDataElement': |
| 558 | """Convert all tensors to np.ndarray in data.""" |
| 559 | new_data = self.new() |
| 560 | for k, v in self.items(): |
| 561 | if isinstance(v, (torch.Tensor, BaseDataElement)): |
| 562 | v = v.detach().cpu().numpy() |
| 563 | data = {k: v} |
| 564 | new_data.set_data(data) |
| 565 | return new_data |
| 566 | |
| 567 | def to_tensor(self) -> 'BaseDataElement': |
| 568 | """Convert all np.ndarray to tensor in data.""" |