使用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))