TensorDataset

TensorDataset是PyTorch中的一个数据集类，用于处理张量数据。它将多个张量组合成一个数据集，并将其作为输入提供给模型进行训练或测试。TensorDataset接受多个张量作为输入，并将它们沿着第一个维度进行拼接，即将它们组合成一个大的张量。然后，每个大张量的第i个元素将被视为一个样本，其中每个元素代表该样本在各个张量中对应位置的值。在训练模型时，可以使用TensorDataset和DataLoader两个类来加载和处理数据集。