【pytorch】torch.utils.data.TensorDataset()原版与新版的差异

使用时发现出错了，如下： 原因是新版把之前的data_tensor 和target_tensor去掉了，输入变成了可变参数，也就是我们平常使用*args class TensorDataset(Dataset): """Dataset wrapping tensors. Each sample will be retrieved by indexing tensors along

>>阅读原文<<