search

xiaobaoonline / pytorch-in-action

MIT License
173 stars 79 forks source link

更新：torch.utils.data.TensorDataset()函数 #16

Closed ghost closed 4 years ago

ghost commented 4 years ago

原版使用方法

train_dataset = Data.TensorDataset(data_tensor=x, target_tensor=y)

新版使用方法

train_dataset = Data.TensorDataset(x,y)

jiangyangbo commented 4 years ago

不错，谢谢！