数据集打包—TensorDataset TensorDataset 可以用来对 tensor 进行打包,就好像 python 中的 zip 功能。该类通过每一个 tensor 的第一个维度进行索引。因此,该类中的 tensor 第一维度必须相等。 from torch.utils.data import TensorDataset import torch from torch.utils.data import DataLoader data_a = torch.tensor( [[11, 22, 33], [44, 55, 66], [77, 88, 99],