
374 Глава 12
Мы можем легко перебирать записи набора данных следующим образом:
>>> for item in data_loader:
... print(item)
tensor([0.])
tensor([1.])
tensor([2.])
tensor([3.])
tensor([4.])
tensor([5.])
Допустим, нам нужно создать из этого набора данных пакеты размером
3
. Воспользу-
емся аргументом
batch_size
в следующем коде:
>>> data_loader = DataLoader(t, batch_size=3, drop_last=False)
>>> for i, batch in enumerate(data_loader, 1):
... print(f'batch {i}:', batch)
batch 1: tensor([0., 1., 2.])
batch 2: tensor([3., 4., 5.])
Этот код создает из текущего набора данных два пакета, где первые три элемента пе-
рейдут в пакет
#1
, а остальные