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