pytorch中使用torch.utils.data.TensorDataset时报错TypeError: 'int' object is not callable,同时在代码中并没有与TensorDataset重名的函数的解决办法。

使用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))
原文地址:https://www.cnblogs.com/wyhluckdog/p/11842285.html