class torch.utils.data.TensorDataset(data_tensor, target_tensor)
原文:https://www.cnblogs.com/hahaah/p/14908633.html