使用TensorDataset函数的代码为:

train_dataset = Data.TensorDataset(x_train,y_train)

执行之后发现报错:

TypeError: 'int' object is not callable。但是检查代码发现并没有与TensorDataset重名的函数。

经过研究TensorDataset函数的源码发现,这个函数传入的参数必须是tensor类型的,所以把x_train与y_train转换为tensor类型在执行这个函数就不报错了,更改后的代码为:

train_dataset = Data.TensorDataset(pt.tensor(x_train),pt.tensor(y_train))

 

posted on 2019-11-12 15:15  wyhluckydog  阅读(4270)  评论(0编辑  收藏  举报