1. from torch.utils.data import Dataset, DataLoader
    4. class Example:
    5.     def __init__(self):
    6.         self.gg = 'hhh'
    7.         self.hh = [2, 2, 2]
    10. class test_dataset(Dataset):
    11.     def __init__(self):
    12.         self.data=[]
    13.         for i in range(16):
    14.             self.data.append(Example())
    16.     def __len__(self):
    17.         return len(self.data)
    19.     def __getitem__(self, i):
    20.         return self.data[i]
    22. def collate_fn(batch):
    23.     data1 = []
    24.     data2 = []
    26.     for i in batch:
    27.         data1.append(i.gg)
    28.         data2.append(i.hh)
    29.     return data1,data2
    31. data = test_dataset()
    32. loader = DataLoader(data, batch_size=1, shuffle=False,collate_fn=collate_fn)
    33. for x,y in loader:
    34.     print(x)
    35.     print(y)
    36.      exit()
    40. output:
    41. ['hhh']
    42. [[2, 2, 2]]