pytorch学习笔记六：torch.utils.data下的TensorDataset和DataLoader的使用

一、TensorDataset

• 对给定的tensor数据(样本和标签)，将它们包装成dataset。注意，如果是numpy的array，或者Pandas的DataFrame需要先转换成Tensor。

'''
data_tensor (Tensor) －　样本数据
target_tensor (Tensor) －　样本目标（标签）
'''
 dataset=torch.utils.data.TensorDataset(data_tensor, 
                                        target_tensor)

下面举个例子：

• 我们先定义一下样本数据和标签数据，一共有1000个样本