Pytorch数据处理流程理解(dataset、dataloader)


torch.Tensor

view(维度):返回一个和原有tensor相同数据,但大小会根据参数进行变形的新Tensor,(-1,x):-1则表示这一维由其它维度推出,可以通过.size()查看Tensor维度

torch.utils.data.DataLoader:读取构造 batch 训练数据
set_epoch:改变每次epoch的index,保证新版pytorch中shuffle参数的兼容

dataset
batch_size
shuffle
sample
num_workers
collate_fn