PyTorch Dataset
A dataset is a Python object that can retrieve an data item by index, and can also return the size of the dataset. The following is sample code that inplements a dataset, with the functionality of each function missing.
class Dataset(obj):
def __getitem__(self, index):
pass
def __len__(self):
pass
pass