Source code for draugr.torch_utilities.datasets.random_dataset
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from typing import Any, Sequence
import torch
from torch.utils.data import Dataset
__author__ = "Christian Heider Nielsen"
__doc__ = r"""
Created on 09/10/2019
"""
__all__ = ["RandomDataset"]
[docs]class RandomDataset(Dataset):
"""description"""
[docs] def __init__(self, nd_size: Sequence, length: int):
self.len = length
self.data = torch.randn((length, *nd_size))
def __getitem__(self, index: int) -> Any:
return self.data[index]
def __len__(self):
return self.len
if __name__ == "__main__":
print(RandomDataset((5, 5), 10)[0])