data.TensorDataset解析

最新推荐文章于 2025-02-05 18:14:29 发布

原创最新推荐文章于 2025-02-05 18:14:29 发布 · 1.2k 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#Pytorch

本文介绍了PyTorch中的TensorDataset类，它用于组合多个张量构成数据集，便于训练时的数据加载和迭代。通过示例展示了如何用特征张量和标签张量创建及访问TensorDataset对象。

部署运行你感兴趣的模型镜像

data.TensorDataset 是 PyTorch 中的一个类，用于创建一个包含多个张量的数据集。这个类的主要作用是将输入的张量组合成一个数据集，使得在训练过程中可以方便地进行数据加载和迭代。

具体来说，TensorDataset 接受一系列的张量作为输入参数，并且将这些张量作为数据集的元素。在实际应用中，通常将特征张量和标签张量作为输入，每个样本的特征和标签分别对应一个位置上的张量。

下面是一个简单的例子，说明如何使用 TensorDataset：shu

import torch
from torch.utils.data import TensorDataset

# 假设有特征张量 features 和标签张量 labels
features = torch.tensor([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]])
labels = torch.tensor([0, 1, 0])

# 使用 TensorDataset 创建数据集
dataset = TensorDataset(features, labels)

# 可以通过索引访问数据集中的元素
sample = dataset[0]
print("First sample:", sample)
print("Second sample:", dataset[1])

输出：