- 先导入数据:
将要处理的文件数据导入x_train,y_train,x_valid,y_valid中,转换为张量。
使用PyTorch中的数据模块,将上面转换为张量的数据使用TensorDataset函数转换为数据集。
之后使用DataLoader来加载数据集,DataLoader包相当于一个派发器的感觉。
下方也是一个导入数据示例但是这是直接从PyTorch中的包含的datasets包中下载的数据并读取转换为张量,再放入加载器中,batch_size就是mini-batch操作每一批迭代取多少个数据。shuffle是随机加载的意思。
标签:转换,张量,PyTorch,导入,数据,加载 From: https://www.cnblogs.com/qianzhu/p/18393489