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])