Data.TensorDataset(data_tensor=x,target_tensor=y) TypeError: __init__() got an unexpected keyword

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

在学习pytorch批处理的时候使用上述代码短会出错:

Traceback (most recent call last):
File “E:/pyproject/pytorchlesson/pytorth_learn_batch_training.py”, line 10, in
torch_dataset=Data.TensorDataset(data_tensor=x,target_tensor=y)
TypeError: init() got an unexpected keyword argument ‘data_tensor’

这是由于pytorch的版本可能跟自己下载的不同导致的,在这里插入图片描述
高版本中直接传入参数即可,改成如下代码即可运行:

torch_dataset=Data.TensorDataset(x,y)

你可能感兴趣的:(学习,python,python常见错误解决,pytorch,深度学习,python)