【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

>>阅读原文<<