dataloader传入tensor
时间: 2023-10-03 13:01:48 浏览: 52
PyTorch实现重写/改写Dataset并载入Dataloader
dataloader可以传入tensor,需要将tensor转换为Dataset对象。可以使用torch.utils.data.TensorDataset将tensor转换为Dataset对象,然后将Dataset对象传递给dataloader。例如:
```
import torch.utils.data as data_utils
# 创建一个tensor
tensor_data = torch.tensor([[0, 1], [2, 3], [4, 5], [6, 7]])
# 将tensor转换为Dataset对象
dataset = data_utils.TensorDataset(tensor_data)
# 创建dataloader
dataloader = data_utils.DataLoader(dataset, batch_size=2, shuffle=True)
# 遍历dataloader
for batch in dataloader:
print(batch)
```
在这个例子中,我们首先创建了一个tensor,然后使用TensorDataset将其转换为Dataset对象。接着,我们创建了一个dataloader,并将Dataset对象传递给它。最后,我们遍历dataloader并打印每个batch。
阅读全文