本文共 618 字，大约阅读时间需要 2 分钟。

学习莫烦pytorch视频，部分函数写法进行更新

import torchimport torch.utils.data as DataBATCH_SIZE=8x=torch.linspace(1,10,10)y=torch.linspace(10,1,10)torch_dataset = Data.TensorDataset(x, y)#新版本直接用元组，(data_tensor=x, target_tensor=y)loader = Data.DataLoader(    dataset = torch_dataset,    batch_size = BATCH_SIZE,    shuffle=False, #是否需要随机打乱True:打乱    num_workers=2,#2个进程或者线程#不够整除)#变成一小批一小批的if __name__=='__main__':##加上这个才能开多线程    for epoch in range(3):        for step, (batch_x, batch_y) in enumerate(loader):            #train            print('Epoch:',epoch,'|step: ',step, '|batch x:', batch_x.numpy(), '|batch y: ', batch_y.numpy())

转载地址：http://owksi.baihongyu.com/