python - 匹配自定义数据集的 PyTorch 张量维度

Question

我对 PyTorch 张量有一些问题，生成的数据集包含4 个维度，我正在尝试将我的自定义数据集与之匹配。因此，我编写了这段代码：

tensorDataset = torch.FloatTensor(random.sample(train_set, 4))
train_loader = torch.utils.data.DataLoader(tensorDataset, batch_size=batch_size, shuffle=True,)

其中train_set是 6 个列表的列表，颜色值为 256 像素（按顺序）且batch_size= 32

我认为在尝试枚举 train_loader 时会发生错误：for n, real_samples in enumerate(train_loader)

在这里，real_samples只接收到想要的4的3 个维度，这会造成手头的问题。

提前致谢！

标签： pythonneural-networkpytorch