仍然报错AttributeError: 'TensorDataset' object has no attribute 'DataLoader'
时间: 2023-09-15 14:15:14 浏览: 308
python的mysql数据查询及报错AttributeError: ‘Connection’ object has no attribute ‘curson’
这个错误是因为`TensorDataset`对象没有`DataLoader`属性。你需要使用`torch.utils.data.DataLoader`来创建数据加载器。下面是一个示例代码:
```python
import torch
from torch.utils.data import DataLoader, TensorDataset
# 创建数据集
x = torch.randn(100, 10)
y = torch.randn(100, 1)
dataset = TensorDataset(x, y)
# 创建数据加载器
batch_size = 10
dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True)
# 使用数据加载器迭代数据
for inputs, targets in dataloader:
# 在这里对每个批量进行处理
pass
```
请注意,`DataLoader`对象本身不是数据集,而是用于迭代数据集的迭代器。因此,您需要使用`for`循环遍历`DataLoader`对象,以获取每个批次的数据。
阅读全文