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

耐心的小黑

于 2020-04-08 14:25:30 发布

阅读量1.7w

点赞数 35

分类专栏： # PyTorch学习笔记文章标签：深度学习 pytorch

本文链接：https://blog.youkuaiyun.com/qq_39507748/article/details/105385709

版权

17 篇文章 ¥19.90 ¥99.00

订阅专栏

本文介绍了如何使用PyTorch的TensorDataset和DataLoader。TensorDataset用于将样本和标签数据转化为Tensor，而DataLoader提供单进程或多进程的数据迭代，包括不同类型的采样方式。文中通过实例展示了它们的用法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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

了解本专栏