從零開(kāi)始深度學(xué)習(xí)Pytorch筆記(11)—— DataLoader類

前文傳送門：從零開(kāi)始深度學(xué)習(xí)Pytorch筆記（1）——安裝Pytorch從零開(kāi)始深度學(xué)習(xí)Pytorch筆記（2）——張量的創(chuàng)建（上）從零開(kāi)始深度學(xué)習(xí)Pytorch筆記（3）——張量的創(chuàng)建（下）從零開(kāi)始深度學(xué)習(xí)Pytorch筆記（4）——張量的拼接與切分從零開(kāi)始深度學(xué)習(xí)Pytorch筆記（5）——張量的索引與變換從零開(kāi)始深度學(xué)習(xí)Pytorch筆記（6）——張量的數(shù)學(xué)運(yùn)算從零開(kāi)始深度學(xué)習(xí)Pytorch筆記（7）——?使用Pytorch實(shí)現(xiàn)線性回歸從零開(kāi)始深度學(xué)習(xí)Pytorch筆記（8）——?計(jì)算圖與自動(dòng)求導(dǎo)（上）
從零開(kāi)始深度學(xué)習(xí)Pytorch筆記（9）——?計(jì)算圖與自動(dòng)求導(dǎo)（下）從零開(kāi)始深度學(xué)習(xí)Pytorch筆記（10）—— Dataset類

DataLoader(dataset,?batch_size=1,?shuffle=False,?sampler=None,?batch_sampler=None,?num_workers=0,?collate_fn=None,?pin_memory=False,?drop_last=False,?timeout=0,?worker_init_fn=None,?multiprocessing_context=None)

import?torch
import?torch.utils.data?as?Data

BATCH_SIZE?=?5

x?=?torch.linspace(1,?20,?20)
y?=?torch.linspace(20,?1,?20)

torch_dataset?=?Data.TensorDataset(x,?y)#創(chuàng)建Dataset

loader?=?Data.DataLoader(
????dataset=torch_dataset,#數(shù)據(jù)
????batch_size=BATCH_SIZE,#批次的大小
????shuffle=True,#打亂數(shù)據(jù)
????num_workers=2,#多進(jìn)程
)

for?epoch?in?range(3):#三個(gè)epoch
????for?step,?(batch_x,?batch_y)?in?enumerate(loader):
????????print('Epoch:?',?epoch,?'|?Step:?',?step,?'|?batch?x:?',
??????????????batch_x.numpy(),?'|?batch?y:?',?batch_y.numpy())

import?torch
import?torch.utils.data?as?Data

BATCH_SIZE?=?4

x?=?torch.linspace(1,?20,?20)
y?=?torch.linspace(20,?1,?20)

torch_dataset?=?Data.TensorDataset(x,?y)#創(chuàng)建Dataset

loader?=?Data.DataLoader(
????dataset=torch_dataset,#數(shù)據(jù)
????batch_size=BATCH_SIZE,#批次的大小
????shuffle=False,#打亂數(shù)據(jù)
????num_workers=2,#多進(jìn)程
)

for?epoch?in?range(3):#三個(gè)epoch
????for?step,?(batch_x,?batch_y)?in?enumerate(loader):
????????print('Epoch:?',?epoch,?'|?Step:?',?step,?'|?batch?x:?',
??????????????batch_x.numpy(),?'|?batch?y:?',?batch_y.numpy())