PyTorch:torch.utils.data

表示Dataset的抽象类。所有其他数据集都应该进行子类化。所有子类应该override__len__和__getitem__,前者提供了数据集的大小,后者支持整数索引,范围从0到len(self)。包装数据和目标张量的数据集。通过沿着第一个维度索引两个张量来恢复每个样本。参数:data_tensor (Tensor) - 包含样本数据target_tensor (Tensor) - 包含样本目标