PyTorch 1.0 中文文档：torch.utils.data

译者：BXuan694

class torch.utils.data.Dataset

表示数据集的抽象类。

所有用到的数据集都必须是其子类。这些子类都必须重写以下方法：__len__：定义了数据集的规模；__getitem__：支持0到len(self)范围内的整数索引。

class torch.utils.data.TensorDataset(*tensors)

用于张量封装的Dataset类。

张量可以沿第一个维度划分为样例之后进行检索。

class torch.utils.data.ConcatDataset(datasets)

用于融合不同数据集的Dataset类。目的：组合不同的现有数据集，鉴于融合操作是同时执行的，数据集规模可以很大。

class torch.utils.data.Subset(dataset, indices)

用索引指定的数据集子集。

参数：

阅读全文／改进本文

转载于:https://www.cnblogs.com/wizardforcel/p/10492627.html