torch_MyDataset.py

import torch.utils.data as data
class MyDataset(data.Dataset):
    '''
    Dataset must define __getitem__ and __len__
    '''
    def __init__(self, others):
        pass

    def __getitem__(self, index):
        """
        index位置的(x, y), x和y都是tensor
        Returns one data pair (x and y).
        """
        return x, y

    def __len__(self):
        # 0<= index < lens
        return lens