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