函数解答:
1、data.TensorDataset可以用来对tensor进行打包。
更明白点来说就是将样本和lable合并起来(样本的个数于lable的长度必须一致)
input:连个相同长度的tensor
output:打包好的tensor
example:
如何显示打包后的tensor:
for x_train, y_label in train_ids: 对样本和lable分开输出
next(iter(data_iter)):打包好的数据整体输出
2、 data.DataLoader(dataset, batch_size, shuffle=is_train)
对data.TensorDataset打包好的数据进行一个选取。
batch_size=选取样本的个数
shuffle=true随机选取,false顺序选取