拼接数据集与向量 需要把torch的一位向量拼接到数据集中,使得遍历每个batch的时候也能遍历这个向量 # 先转化为TensorDataset的数据集格式 xin_onehot = Data.TensorDataset(xin_onehot) #正常的数据集 data_loader = DataLoader(dataset=cifar10, batch_size=32, shuffle=False) # 需要加入的1位向量 xin_onehot = Data