擅长:python、mysql、java
<p>我认为DataLoader实际需要的是一个子类<code>Dataset</code>的输入。您可以编写自己的数据集类,使其成为<code>Dataset</code>的子类,也可以使用<code>TensorDataset</code>,如下所示:</p>
<pre><code>import torch
import numpy as np
from torch.utils import data
my_x = [np.array([[1.0,2],[3,4]]),np.array([[5.,6],[7,8]])] # a list of numpy arrays
my_y = [np.array([4.]), np.array([2.])] # another list of numpy arrays (targets)
tensor_x = torch.Tensor(my_x) # transform to torch tensor
tensor_y = torch.Tensor(my_y)
my_dataset = data.TensorDataset(tensor_x,tensor_y) # create your datset
my_dataloader = data.DataLoader(my_dataset) # create your dataloader
</code></pre>
<p>对我有用。希望对你有帮助。</p>