| import torch | |
| device = torch.device("cuda") | |
| tenz = torch.tensor([1.,2.], device=device) | |
| #tenz.toDevice(device) | |
| print(torch.cuda.is_available()) | |
| from datasets import Dataset | |
| dataset = Dataset.from_dict({"a": [0, 1, 2]}) | |
| dataset_with_duplicates = dataset.map(lambda batch: {"b": batch["a"] * 2}) | |
| print(dataset_with_duplicates.shape) | |
| len(dataset_with_duplicates) | |
| dataset_with_duplicates[:] |